Skip to content

Commit b15966e

Browse files
committed
Use n_cores jobs in GridSearchCV
1 parent bfca4eb commit b15966e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

cesium_app/handlers/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ def _build_model_compute_statistics(fset_path, model_type, model_params,
6262
raise ValueError("Cannot build model for unlabeled feature set.")
6363
model = MODELS_TYPE_DICT[model_type](**model_params)
6464
if params_to_optimize:
65-
model = GridSearchCV(model, params_to_optimize)
65+
model = GridSearchCV(model, params_to_optimize, n_jobs=-1)
6666
model.fit(fset, data['labels'])
6767
score = model.score(fset, data['labels'])
6868
best_params = model.best_params_ if params_to_optimize else {}

0 commit comments

Comments
 (0)