diff --git a/experimentsClassicClassifiers.py b/experimentsClassicClassifiers.py index 3c6991549d1b2464c614f22725e3892c66ac7eb8..fd2558cf8cc5436c7ca2906833ee9a2bb7c23efc 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)