Update PCRA.py

apoorvumang · Aug 30, 2018 · 0120861 · 0120861
1 parent b48738a
commit 0120861
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/PTransE/PCRA.py b/PTransE/PCRA.py
@@ -23,11 +23,8 @@ def map_add1(mp,key):
     seg = line.strip().split()
     relation2id[seg[0]] = int(seg[1])
     id2relation[int(seg[1])]=seg[0]
+    id2relation[int(seg[1])+ relation_num]="~"+seg[0]
     relation_num+=1
-
-for line in f:
-    seg = line.strip().split()
-    id2relation[int(seg[1])+relation_num]="~"+seg[0]
 f.close()
 
 ok = {}