Merge pull request #16 from kumo-ai/yyuan/include-shallowrhs-to-bench…

…mark fix code based on review
kumo-ai · Aug 2, 2024 · 1a336eb · 1a336eb
2 parents 78e11cd + 4c07865
commit 1a336eb
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -5,6 +5,7 @@ __pycache__/
 build/
 dist/
 alpha/
+result/
 .cache/
 .eggs/
 *.egg-info/
@@ -17,4 +18,4 @@ coverage.xml
 .venv
 venv/*
 *.out
-data/**
+data/**
diff --git a/benchmark/relbench_link_prediction_benchmark.py b/benchmark/relbench_link_prediction_benchmark.py
@@ -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)

diff --git a/result/rel-stack_user-post-comment_hybridgnn b/result/rel-stack_user-post-comment_hybridgnn
-Original file 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/**