From 14877937d89d8a49117bd402b9822009da53f54c Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Mon, 30 Sep 2024 11:25:29 +0200
Subject: [PATCH] fix save preds with file
---
main_custom.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/main_custom.py b/main_custom.py
index 994b601..7de21bd 100644
--- a/main_custom.py
+++ b/main_custom.py
@@ -346,7 +346,8 @@ def save_pred(model, data_val, forward, output_path, file = False):
data_frame['pred int'] = pred_int
data_frame['true int'] = true_int
data_frame['charge'] = charges
- data_val.data.set_file_mode(False)
+ if file :
+ data_val.dataset.set_file_mode(False)
data_frame.to_csv(output_path)
--
GitLab