From e1944e193122a1adbc99a66c331a547c92437b84 Mon Sep 17 00:00:00 2001
From: celine <celine.robardet@insa-lyon.fr>
Date: Wed, 14 Feb 2024 13:12:11 +0100
Subject: [PATCH] indentation
---
code/binary_bpr_ablation/script_ablation.py | 4 ++--
code/nary_model/main_nary_cv.py | 2 +-
2 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/code/binary_bpr_ablation/script_ablation.py b/code/binary_bpr_ablation/script_ablation.py
index 5e56aca..9c957a2 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 609ed98..6816f1d 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):
--
GitLab