Skip to content

Commit

Permalink
Update transR.py
Browse files Browse the repository at this point in the history
  • Loading branch information
guotong1988 authored Nov 12, 2019
1 parent 1f2ce7e commit 24ab446
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions transR.py
Original file line number Diff line number Diff line change
Expand Up @@ -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.0001
self.testFlag = False
Expand Down

0 comments on commit 24ab446

Please sign in to comment.