Parcourir la source

add parallelization for computation

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

+ 2 - 2
models.py

@@ -28,7 +28,7 @@ def _get_best_model(X_train, y_train):
     param_grid = {'kernel':['rbf'], 'C': Cs, 'gamma' : gammas}
     param_grid = {'kernel':['rbf'], 'C': Cs, 'gamma' : gammas}
 
 
     svc = svm.SVC(probability=True, class_weight='balanced')
     svc = svm.SVC(probability=True, class_weight='balanced')
-    clf = GridSearchCV(svc, param_grid, cv=10, verbose=1, scoring=my_accuracy_scorer)
+    clf = GridSearchCV(svc, param_grid, cv=10, verbose=1, scoring=my_accuracy_scorer, n_jobs=-1)
 
 
     clf.fit(X_train, y_train)
     clf.fit(X_train, y_train)
 
 
@@ -48,7 +48,7 @@ def _get_best_gpu_model(X_train, y_train):
     param_grid = {'kernel':['rbf'], 'C': Cs, 'gamma' : gammas}
     param_grid = {'kernel':['rbf'], 'C': Cs, 'gamma' : gammas}
 
 
     svc = SVC(probability=True, class_weight='balanced')
     svc = SVC(probability=True, class_weight='balanced')
-    clf = GridSearchCV(svc, param_grid, cv=10, verbose=1, scoring=my_accuracy_scorer)
+    clf = GridSearchCV(svc, param_grid, cv=10, verbose=1, scoring=my_accuracy_scorer, n_jobs=-1)
 
 
     clf.fit(X_train, y_train)
     clf.fit(X_train, y_train)