Parcourir la source

reduce of number of params

Jérôme BUISINE il y a 2 ans
Parent
commit
65849b6228
1 fichiers modifiés avec 3 ajouts et 3 suppressions
  1. 3 3
      models.py

+ 3 - 3
models.py

@@ -25,13 +25,13 @@ from sklearn.model_selection import KFold, cross_val_score
 
 def _get_best_model(X_train, y_train):
 
-    Cs = [0.001, 0.01, 0.1, 1, 10, 100, 1000]
-    gammas = [0.001, 0.01, 0.1, 5, 10, 100]
+    Cs = [0.01, 0.1, 10, 100, 1000]
+    gammas = [0.01, 0.1, 10, 100]
     param_grid = {'kernel':['rbf'], 'C': Cs, 'gamma' : gammas}
 
     svc = svm.SVC(probability=True, class_weight='balanced')
     #clf = GridSearchCV(svc, param_grid, cv=5, verbose=1, scoring=my_accuracy_scorer, n_jobs=-1)
-    clf = GridSearchCV(svc, param_grid, cv=5, verbose=1, n_jobs=-1)
+    clf = GridSearchCV(svc, param_grid, cv=3, verbose=1, n_jobs=-1)
 
     clf.fit(X_train, y_train)