Parcourir la source

Update CV parameter when training

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

+ 1 - 1
models.py

@@ -31,7 +31,7 @@ def _get_best_model(X_train, y_train):
 
     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=3, verbose=1, n_jobs=-1)
+    clf = GridSearchCV(svc, param_grid, cv=5, verbose=1, n_jobs=-1)
 
     clf.fit(X_train, y_train)