Skip to content
Snippets Groups Projects
Commit 14877937 authored by Schneider Leo's avatar Schneider Leo
Browse files

fix save preds with file

parent f6a636e6
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
Showing
with 2 additions and 1 deletion
main_custom.py
+ 2
− 1
View file @ 14877937
...@@ -346,7 +346,8 @@ def save_pred(model, data_val, forward, output_path, file = False): ...@@ -346,7 +346,8 @@ def save_pred(model, data_val, forward, output_path, file = False):
data_frame['pred int'] = pred_int data_frame['pred int'] = pred_int
data_frame['true int'] = true_int data_frame['true int'] = true_int
data_frame['charge'] = charges 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) data_frame.to_csv(output_path)
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment