Merge pull request #75 from guoci/workaround_loky_issue

workaround `loky` issue
grosenberger · Feb 23, 2022 · 4a85149 · 4a85149
2 parents 15b9aee + 0126398
commit 4a85149
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/easypqp/library.py b/easypqp/library.py
@@ -211,11 +211,10 @@ def predict(self, x):
   def __repr__(self):
     return str(self.get_params())
 
-
 def lowess_iso_predictor(filename, x, y, xpred):
   gsc = sklearn.model_selection.GridSearchCV(LowessIsoEstimator(None), {'lowess_frac': [0.01, 0.02, 0.04, 0.08]},
                                              cv=sklearn.model_selection.KFold(4, shuffle=True, random_state=0),
-                                             n_jobs=-1)
+                                             n_jobs=min(os.cpu_count(), 61))
 
   gsc.fit(x.reshape(-1, 1), y)
   click.echo(f'Info: {filename}; Lowess fraction used: {gsc.best_params_["lowess_frac"]}.')