Parcourir la source

custom search param

Jérôme BUISINE il y a 3 ans
Parent
commit
5df4f32b6c
1 fichiers modifiés avec 9 ajouts et 4 suppressions
  1. 9 4
      models.py

+ 9 - 4
models.py

@@ -54,21 +54,26 @@ def _get_best_gpu_model(X_train, y_train):
     gammas = [0.001, 0.01, 0.1, 5, 10, 100]
 
     bestModel = None
-    modelScore = 0.
+    bestScore = 0.
+
+    n_eval = 1
 
     for c in Cs:
         for g in gammas:
 
-            print('C:', c, ', gamma:', g)
             svc = SVC(probability=True, class_weight='balanced', kernel='rbf', gamma=g, C=c)
             svc.fit(X_train, y_train)
 
             score = svc.score(X_train, y_train)
 
-            if score > modelScore:
-                modelScore = score
+            # keep track of best model
+            if score > bestScore:
+                bestScore = score
                 bestModel = svc
 
+            print('Eval n° {} [C: {}, gamma: {}] => [score: {}, bestScore:{}]'.format(n_eval, c, g, score, bestScore))
+            n_eval += 1
+
     return bestModel
 
 def svm_gpu(X_train, y_train):