Used all the core available when fine tuning

adamspd · Jan 20, 2024 · 0503308 · 0503308
1 parent 6995a05
commit 0503308
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/spam_detector_ai/test_and_tuning/fine_tuning_svm.py b/spam_detector_ai/test_and_tuning/fine_tuning_svm.py
@@ -17,7 +17,7 @@
                   'kernel': ['rbf', 'linear', 'poly', 'sigmoid']}
 
     logger.info("Define the grid search")
-    grid_search = GridSearchCV(SVC(), param_grid, refit=True, verbose=3, cv=5)
+    grid_search = GridSearchCV(SVC(), param_grid, refit=True, verbose=3, cv=5, n_jobs=-1)
 
     logger.info("Loading the training data")
     trainer = ModelTrainer(data_path='../data/spam.csv', classifier_type=ClassifierType.SVM, logger=logger)