fix transae bugs

M-KGE/IKRL_TransAE/TransAE.py

@@ -986,14 +986,14 @@ def create_mappings(dataset_path: str):
             transe = TransE(
                 ent_tot = len(entity2id),
                 rel_tot = len(relation2id),
-                dim = 400, 
+                dim = 200, 
                 p_norm = 1, 
                 norm_flag = True)
         else:
             transe = Analogy(
                 ent_tot = len(entity2id),
                 rel_tot = len(relation2id),
-                dim = 400, 
+                dim = 200, 
                 p_norm = 1, 
                 norm_flag = True)
 
@@ -1005,10 +1005,10 @@ def create_mappings(dataset_path: str):
 
         trainer = Trainer(model=model, data_loader=train_dataloader, train_times=2000, alpha=alpha, use_gpu=True)
         trainer.run()
-        transe.save_checkpoint('ckpt/analogy/pt_trainse.ckpt')
+        transe.save_checkpoint('ckpt/analogy/pt_transe.ckpt')
 
         # test the model
-        transe.load_checkpoint('ckpt/analogy/pt_trainse.ckpt')
+        transe.load_checkpoint('ckpt/analogy/pt_transe.ckpt')
         tester = Tester(model=transe, data_loader= test_dataloader, use_gpu = True)
         tester.run_link_prediction(type_constrain = False)
 
@@ -1066,4 +1066,4 @@ def create_mappings(dataset_path: str):
         print("hit10: ", hit10)
         print("hti5: ", hti5)
         print("hti3: ", hti3)
-        print("hit1: ", hit1)
+        print("hit1: ", hit1)