From a94410b71efe2b22430f9987d6a8dc4a604670ef Mon Sep 17 00:00:00 2001 From: shaneing Date: Wed, 8 Jul 2020 23:31:52 +0800 Subject: [PATCH] fix: create output directory if not exists --- utils/IOUtil.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/utils/IOUtil.py b/utils/IOUtil.py index 118523d..75a42a1 100644 --- a/utils/IOUtil.py +++ b/utils/IOUtil.py @@ -4,6 +4,7 @@ Created on Sun Apr 15 09:49:35 2018 @author: thieunv """ +import os import numpy as np import pandas as pd @@ -19,6 +20,9 @@ def save_loss_train_to_csv(error=None, filename=None, pathsave=None): return None def save_all_models_to_csv(item=None, log_filename=None, pathsave=None): + if not os.path.exists(pathsave): + os.makedirs(pathsave) + with open(pathsave + log_filename + ".csv", "a+") as file: wr = csv.writer(file, dialect='excel') wr.writerow(item)