change more hardcoded CPU counts to os.cpu_count()

This commit is contained in:
Tanushree Tunstall 2022-07-02 10:25:13 +01:00
parent b8653c6afe
commit 2fda32901b
4 changed files with 4 additions and 4 deletions

View file

@ -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)