Update scripts.

DeepGraphLearning · Jul 30, 2021 · 318d1f5 · 318d1f5
1 parent 69aafd4
commit 318d1f5
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/codes/extra/core/__init__.py b/codes/extra/core/__init__.py
@@ -0,0 +1 @@
+from .base import _MetaContainer, Registry, Configurable
diff --git a/codes/extra/utils/__init__.py b/codes/extra/utils/__init__.py
@@ -0,0 +1,4 @@
+from .io import input_choice, literal_eval
+from .file import download, extract, compute_md5, get_line_count
+from .torch import cpu, cuda
+from . import pretty
diff --git a/codes/run.py b/codes/run.py
@@ -16,7 +16,7 @@
 Example:
 The script will train a separate model for each relation in range(start, total number of relations, hop).
 
-DATA_DIR             = "../data/kinship_325"
+DATA_DIR             = "../data/kinship"
 OUTPUT_DIR           = "./workspace"
 write_log_to_console = True
 start                = 0