Przeglądaj źródła

Merge branch 'release/v0.0.9'

Jérôme BUISINE 4 lat temu
rodzic
commit
62fc57be97
1 zmienionych plików z 3 dodań i 0 usunięć
  1. 3 0
      train_model.py

+ 3 - 0
train_model.py

@@ -63,6 +63,9 @@ def train(_data_file, _model_name):
 
     dump(clf, model_file_path)
 
+    if not os.path.exists(cfg.results_information_folder):
+        os.makedirs(cfg.results_information_folder)
+        
     # save score into global_result.csv file
     with open(cfg.global_result_filepath, "a") as f:
        f.write(model_filename.replace('.joblib', '') + ';' + str(len(y)) + ';' + str(coeff) + ';\n')