From ac00704640d7fe1f783e3fa39be9f36d4e0764f0 Mon Sep 17 00:00:00 2001
From: lmoncla <ludovic.moncla@insa-lyon.fr>
Date: Mon, 7 Jun 2021 14:48:00 +0200
Subject: [PATCH] udpate
---
experimentsClassicClassifiers.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/experimentsClassicClassifiers.py b/experimentsClassicClassifiers.py
index 3c69915..fd2558c 100644
--- a/experimentsClassicClassifiers.py
+++ b/experimentsClassicClassifiers.py
@@ -99,16 +99,16 @@ for columnInput in [columnText, 'firstParagraph']:
if clf_name != 'bayes' :
clf = GridSearchCV(clf, grid_param, refit = True, verbose = 3)
- else if feature_technique_name == 'doc2vec':
+ elif feature_technique_name == 'doc2vec':
continue
t_begin = time.time()
- if os.path.isfile(os.path.join('./model', model_file_name)):
- with open(model_file_name, 'rb') as file:
+ if os.path.isfile(os.path.join('./models', model_file_name)):
+ with open(os.path.join('./models', model_file_name), 'rb') as file:
clf = pickle.load(file)
else:
- with open(Pkl_Filename, 'wb') as file:
+ with open(os.path.join('./models', model_file_name), 'wb') as file:
clf.fit(train_x, train_y)
pickle.dump(clf, file)
--
GitLab