Skip to content

Commit

Permalink
fix code based on review
Browse files Browse the repository at this point in the history
  • Loading branch information
yiweny committed Aug 2, 2024
1 parent 47285f4 commit 4c07865
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 2 deletions.
3 changes: 2 additions & 1 deletion .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@ __pycache__/
build/
dist/
alpha/
result/
.cache/
.eggs/
*.egg-info/
Expand All @@ -17,4 +18,4 @@ coverage.xml
.venv
venv/*
*.out
data/**
data/**
2 changes: 1 addition & 1 deletion benchmark/relbench_link_prediction_benchmark.py
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,7 @@
parser.add_argument(
"--model",
type=str,
default="shallowrhsgnn",
default="hybridgnn",
choices=["hybridgnn", "idgnn", "shallowrhsgnn"],
)
parser.add_argument("--epochs", type=int, default=20)
Expand Down
Binary file removed result/rel-stack_user-post-comment_hybridgnn
Binary file not shown.

0 comments on commit 4c07865

Please sign in to comment.