indentation

code/binary_bpr_ablation/script_ablation.py

 import os
 
-name = ["assist09","assist17","algebra", "math1","math2"]
+name = ["assist0910_tkde","assist17_tkde","algebra", "math_1","math_2"]
 for i in range(4):
     print("Ablation (0 no ablation, 1 ablation L2, 2 ablation init, 3 both) ",i)
     for a in range(5):
         print(name[i])
-        cmd = "python main.py --dataTrain ../data/"+name[i]+"/train.csv --dataTest ../data/"+name[i]+"/test.csv --ablation "+str(i)
+        cmd = "python main.py --dataTrain ../../data/"+name[i]+"/train_0.csv --dataTest ../../data/"+name[i]+"/test_0.csv --ablation "+str(i)
         os.system(cmd)  

code/nary_model/main_nary_cv.py

@@ -48,7 +48,7 @@ def evaluate_all(dataTrain, dataTest, filename):
     new_embedding_items = bpr_model.item_embeddings.weight.clone().detach().cpu().numpy()
     write_file(filename+"embedding_items.csv", new_embedding_items[0:nb_item_train])
     # Test
-    acc, precision, rappel, all_decisions = bpr_model.evaluate_model(test, len(dico_kc), y_test) 
+    acc, precision, rappel, all_decisions, all_prefs = bpr_model.evaluate_model(test, len(dico_kc), y_test) 
     '''
     s = str(acc) +","+ str( precision)+ ","+str(rappel)+ ","+str(doa)
     for i in range(embedding_size):