Sfoglia il codice sorgente

remove of custom accuracy score

Jérôme BUISINE 4 anni fa
parent
commit
be3baabba0
1 ha cambiato i file con 8 aggiunte e 7 eliminazioni
  1. 8 7
      models.py

+ 8 - 7
models.py

@@ -16,12 +16,12 @@ from sklearn.model_selection import KFold, cross_val_score
 # variables and parameters
 n_predict = 0
 
-def my_accuracy_scorer(*args):
-        global n_predict
-        score = accuracy_score(*args)
-        print('{0} - Score is {1}'.format(n_predict, score))
-        n_predict += 1
-        return score
+# def my_accuracy_scorer(*args):
+#         global n_predict
+#         score = accuracy_score(*args)
+#         print('{0} - Score is {1}'.format(n_predict, score))
+#         n_predict += 1
+#         return score
 
 def _get_best_model(X_train, y_train):
 
@@ -30,7 +30,8 @@ def _get_best_model(X_train, y_train):
     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, scoring=my_accuracy_scorer, n_jobs=-1)
+    clf = GridSearchCV(svc, param_grid, cv=5, verbose=1, n_jobs=-1)
 
     clf.fit(X_train, y_train)