Quellcode durchsuchen

Update CV parameter when training

Jérôme BUISINE vor 3 Jahren
Ursprung
Commit
62af7d2a9e
1 geänderte Dateien mit 1 neuen und 1 gelöschten Zeilen
  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)