diff --git a/main.py b/main.py index 670705a..0072b7c 100644 --- 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)})