change more hardcoded CPU counts to os.cpu_count()
This commit is contained in:
parent
b8653c6afe
commit
2fda32901b
4 changed files with 4 additions and 4 deletions
|
@ -78,7 +78,7 @@ import itertools
|
|||
from sklearn.model_selection import LeaveOneGroupOut
|
||||
#%% GLOBALS
|
||||
rs = {'random_state': 42}
|
||||
njobs = {'n_jobs': 10}
|
||||
njobs = {'n_jobs': os.cpu_count() } # the number of jobs should equal the number of CPU cores
|
||||
|
||||
scoring_fn = ({ 'mcc' : make_scorer(matthews_corrcoef)
|
||||
, 'fscore' : make_scorer(f1_score)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue