Configuring njobs with Python tpot

Question

Can the njobs parameter be configured using tpot within code and not passed as command line parameter ?

Reading https://rhiever.github.io/tpot/using/ states :

-njobs  NUM_JOBS    Any positive integer or -1  Number of CPUs for evaluating pipelines in parallel during the TPOT optimization process. 

Assigning this to -1 will use as many cores as available on the computer.

But how to configure this parameter within code ?

Trying :

TPOTClassifier(generations=5, verbosity=3,  config_dict='TPOT light' , NUM_JOBS = 4)

returns error :

TPOTClassifier(generations=5, verbosity=3,  config_dict='TPOT light' , NUM_JOBS = 4)
TypeError: __init__() got an unexpected keyword argument 'NUM_JOBS'

score 1 · Accepted Answer · edited Apr 09 '20 at 21:32

1

Achieved using n_jobs parameter:

TPOTClassifier(
    generations=5,
    verbosity=3,
    config_dict='TPOT light',
    n_jobs=4
)

edited Apr 09 '20 at 21:32

gosuto

5,422
6
36
57

answered Oct 27 '17 at 16:09

blue-sky

51,962
152
427
752

Configuring njobs with Python tpot

1 Answers1