test

2024-06-19 00:19:31 +07:00
parent a064d12763
commit 975a0a77c2
1 changed files with 2 additions and 0 deletions
--- a/main.py
+++ b/main.py
@@ -94,6 +94,8 @@ class Main(object):

        # self.ent2id = {ent: idx for idx, ent in enumerate(ent_set)}
        # self.rel2id = {rel: idx for idx, rel in enumerate(rel_set)}
+        
+        print("Num rel1: " + str(len(self.rel2id)))
        self.rel2id.update({rel+'_reverse': idx+len(self.rel2id)
                           for idx, rel in enumerate(rel_set)})