fix rsme

Author: flow3rdown

.gitignore

@@ -10,4 +10,6 @@ training/
 
 *.pkl
 
-*.pth
+*.pth
+
+.DS_Store/*

M-KGE/IKRL_TransAE/DATA/PyTorchTrainDataLoader.py M-KGE/IKRL_TransAE/DATA_/PyTorchTrainDataLoader.py

@@ -11,7 +11,7 @@
 from tqdm import tqdm
 import ctypes
 import pickle
-from DATA import TrainDataLoader, TestDataLoader
+from DATA_ import TrainDataLoader, TestDataLoader
 os.environ['CUDA_VISIBLE_DEVICES'] = '0'
 
 

M-KGE/IKRL_TransAE/DATA/TestDataLoader.py M-KGE/IKRL_TransAE/DATA_/TestDataLoader.py

@@ -14,7 +14,7 @@
 import pickle
 import gensim
 import random
-from DATA import TrainDataLoader, TestDataLoader
+from DATA_ import TrainDataLoader, TestDataLoader
 os.environ['CUDA_VISIBLE_DEVICES'] = '0'
 
 

M-KGE/IKRL_TransAE/DATA/TrainDataLoader.py M-KGE/IKRL_TransAE/DATA_/TrainDataLoader.py

@@ -31,10 +31,10 @@ def extract_feature(self,base_path):
         self.model.eval()
         dict = {}
 
-        file = open('analogy_best_img.pickle', 'rb')
+        file = open('data/analogy/analogy_best_img.pickle', 'rb')
         fb15k_best_img = pickle.load(file)
         file.close()
-        img_paths = list(fb15k_best_img.values())
+        img_paths = [os.path.join(base_path, value) for value in list(fb15k_best_img.values())]
         print(len(img_paths))
         pbart = tqdm(total=len(img_paths))
         while len(img_paths) > 0:
@@ -171,7 +171,7 @@ def save_embedding(self,output_file):
 if __name__ == "__main__":
 
     model1 = VisionTransformer()
-    base_path = '../MarT/dataset/MARS/images/'
+    base_path = 'data/images/'
     model1.get_embedding(base_path)
     model1.save_embedding('analogy_vit_best_img_vec.pickle')
 

M-KGE/IKRL_TransAE/DATA/__init__.py M-KGE/IKRL_TransAE/DATA_/__init__.py

@@ -44,7 +44,7 @@ def get_img_vec_array_forget(proportion,remember_proportion,rank_file='fb15k_vit
                 continue
             rel_rank[rel[10:]]=float(mrp[12:])
 
-    with open('../MarT/dataset/MarKG/relation2text.txt', 'r') as f:
+    with open('../../MarT/dataset/MarKG/relation2text.txt', 'r') as f:
         Lines=f.readlines()
 
     rel_id_pd=[]
@@ -79,7 +79,7 @@ def get_img_vec_sig_alpha(proportion,rank_file='fb15k_vit_rank.txt',eutput_file=
                 continue
             rel_rank[rel[10:]]=float(mrp[12:])
 
-    with open('../MarT/dataset/MarKG/relation2text.txt', 'r') as f:
+    with open('../../MarT/dataset/MarKG/relation2text.txt', 'r') as f:
         Lines=f.readlines()
 
     rel_sig_alpha=[]