Update transE.py

thunlp · Nov 12, 2019 · e642eb4 · e642eb4
1 parent 1f2ce7e
commit e642eb4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/transE.py b/transE.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