Преглед на файлове

fix with use of numpy for mean

Jérôme BUISINE преди 4 години
родител
ревизия
38f647b7fb
променени са 1 файла, в които са добавени 3 реда и са изтрити 0 реда
  1. 3 0
      models.py

+ 3 - 0
models.py

@@ -1,4 +1,6 @@
 # models imports
+import numpy as np
+
 from sklearn.model_selection import GridSearchCV
 from sklearn.linear_model import LogisticRegression
 from sklearn.ensemble import RandomForestClassifier, VotingClassifier
@@ -68,6 +70,7 @@ def _get_best_gpu_model(X_train, y_train):
             svc.fit(X_train, y_train)
 
             score = cross_val_score(svc, X_train, y_train, cv=k_fold, n_jobs=-1)
+            score = np.mean(score)
 
             # keep track of best model
             if score > bestScore: