Skip to content

Commit da6caf0

Browse files
committed
Fix making directories
1 parent ea49a93 commit da6caf0

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

utils/IOUtil.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88
import numpy as np
99
import pandas as pd
1010
import csv
11+
import os
1112

1213
def save_prediction_to_csv(y_test=None, y_pred=None, filename=None, pathsave=None):
1314
t1 = np.concatenate((y_test, y_pred), axis=1)
@@ -19,6 +20,9 @@ def save_loss_train_to_csv(error=None, filename=None, pathsave=None):
1920
return None
2021

2122
def save_all_models_to_csv(item=None, log_filename=None, pathsave=None):
23+
directory = os.getcwd() + "/" + pathsave
24+
if not os.path.exists(directory):
25+
os.makedirs(directory)
2226
with open(pathsave + log_filename + ".csv", "a+") as file:
2327
wr = csv.writer(file, dialect='excel')
2428
wr.writerow(item)

0 commit comments

Comments
 (0)