ML scripts: {'n_jobs': os.cpu_count() }

This commit is contained in:
Tanushree Tunstall 2022-07-02 10:20:40 +01:00
parent 11af00f1db
commit b8653c6afe
4 changed files with 16 additions and 12 deletions

View file

@ -76,7 +76,7 @@ import argparse
import re
#%% 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)