使用 Python tpot配置 njobs

Question

njobs参数是否可以在代码中使用tpot进行配置，而不是作为命令行参数传递？

阅读 https://rhiever.github.io/tpot/using/状态：

-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.

但是如何在代码中配置此参数？

正在尝试 ：

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

返回错误：

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

Answer 1

使用 n_jobs 参数实现：

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