From 5ab8b537dc27fa2018ffca242f66a08ad59e0571 Mon Sep 17 00:00:00 2001
From: Arthur BATEL <arthur.batel@insa-lyon.fr>
Date: Mon, 22 Apr 2024 08:45:44 +0200
Subject: [PATCH] iterations correct
---
code/binary_bpr_ablation/script_ablation.py | 15 ++++++++-------
1 file changed, 8 insertions(+), 7 deletions(-)
diff --git a/code/binary_bpr_ablation/script_ablation.py b/code/binary_bpr_ablation/script_ablation.py
index b7fb6e3..4012781 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)
--
GitLab