diff --git a/code/binary_bpr_ablation/script_ablation.py b/code/binary_bpr_ablation/script_ablation.py
index 5e56acab0d89733e56e7bde3eede2869f04dd323..9c957a2630526f18c8ab6bfaf368262abd549d21 100644
--- a/code/binary_bpr_ablation/script_ablation.py
+++ b/code/binary_bpr_ablation/script_ablation.py
@@ -1,9 +1,9 @@
 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)  
diff --git a/code/nary_model/main_nary_cv.py b/code/nary_model/main_nary_cv.py
index 609ed98d1df6a2b1a7237e1b13951ca1d4415b4d..6816f1d421833630274ad831bffc37d276dc3eac 100644
--- a/code/nary_model/main_nary_cv.py
+++ b/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):