From 22c0fe174a1843fbac3dc92e5c820cfb9c7b8449 Mon Sep 17 00:00:00 2001 From: lmoncla <ludovic.moncla@insa-lyon.fr> Date: Mon, 7 Jun 2021 14:52:20 +0200 Subject: [PATCH] udpate --- classifiers.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/classifiers.py b/classifiers.py index 16db401..f68b2c6 100644 --- a/classifiers.py +++ b/classifiers.py @@ -12,11 +12,11 @@ import numpy as np classifiers = [ ('bayes', MultinomialNB()), + ('lr', LogisticRegression()), + ('sgd', SGDClassifier()), ('svm', SVC() ), ('decisionTree',DecisionTreeClassifier()), ('rfc', RandomForestClassifier()), - ('lr', LogisticRegression()), - ('sgd', SGDClassifier()), ('knn', KNeighborsClassifier()) ] @@ -26,7 +26,7 @@ param_grid_decisionTree = { 'criterion' : ['gini', 'entropy'], 'max_depth':range param_grid_rfc = { 'n_estimators': [200, 500], 'max_features': ['auto', 'sqrt', 'log2'], 'max_depth' : [4,5,6,7,8], 'criterion' :['gini', 'entropy'] } param_grid_lr = {"C":np.logspace(-3,3,7), "penalty":["l1","l2"]} param_grid_sgd = { "loss" : ["hinge", "log", "squared_hinge", "modified_huber"], "alpha" : [0.0001, 0.001, 0.01, 0.1], "penalty" : ["l2", "l1", "none"], "max_iter" : [500]} -param_grid_knn = {'n_neighbors' : list(range(1,20)), 'weights' : ['uniform', 'distance'], 'metric' : ['euclidean', 'manhattan'] } +param_grid_knn = {'n_neighbors' : list(range(3,20)), 'weights' : ['uniform', 'distance'], 'metric' : ['euclidean', 'manhattan'] } grid_params = [ ('bayes', None), -- GitLab