diff --git a/code/binary_bpr_ablation/script_ablation.py b/code/binary_bpr_ablation/script_ablation.py
index b7fb6e3deb0d3bb98f7f71c1d320deeebeaedb35..40127814dfd5bed1b2050099076534fb9cfb2799 100644
--- a/code/binary_bpr_ablation/script_ablation.py
+++ b/code/binary_bpr_ablation/script_ablation.py
@@ -1,10 +1,11 @@
 import os
 
 epochs = [70, 95, 75, 75, 90]
-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 --epochs "+str(epochs[i])+" --dataTrain ../../data/cdbpr_format/"+str(name[i])+"/train_valid_"+str(a)+".csv --dataTest ../../data/cdbpr_format/"+str(name[i])+"/test_"+str(a)+".csv --ablation "+str(i)
-        os.system(cmd)  
+datasets = ["assist0910_tkde", "assist17_tkde", "algebra", "math_1", "math_2"]
+for d,dataset in enumerate(datasets) :
+    print(f'----------{dataset}----------')
+    for i in range(4):
+        print("Ablation (0 no ablation, 1 ablation L2, 2 ablation init, 3 both) ",i)
+        for a in range(5):
+            cmd = "python main.py --epochs " + str(epochs[d]) +" --dataTrain ../../data/cdbpr_format/" + str(dataset) + "/train_valid_" + str(a) + ".csv --dataTest ../../data/cdbpr_format/" + str(dataset) + "/test_" + str(a) + ".csv --ablation " + str(i)
+            os.system(cmd)