Update transH.py

thunlp · Nov 12, 2019 · 7360731 · 7360731
1 parent 1f2ce7e
commit 7360731
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/transH.py b/transH.py
@@ -13,8 +13,9 @@
 class Config(object):
 
 	def __init__(self):
-		lib.setInPath("./data/FB15K/")
-		test_lib.setInPath("./data/FB15K/")
+		path = "./data/FB15K/"
+                lib.setInPath(path, len(path))
+                test_lib.setInPath(path, len(path))
 		lib.setBernFlag(0)
 		self.learning_rate = 0.001
 		self.testFlag = False