sklearn套索会自动并行运行吗?

时间:2018-11-24 02:18:51

标签: python scikit-learn slurm

我想使用sklearn.linear_model.Lasso模块并行运行套索回归。

我用于提交作业的bash脚本就是这样

#!/bin/bash
#SBATCH --partition=the_partition
#SBATCH --nodes=2
#SBATCH --ntasks=56
python lasso_script.py

lasso_script.py中的相关部分是

Lasso(alpha=0.1, copy_X=True, fit_intercept=True, max_iter=1000,
  normalize=False, positive=False, precompute=False, random_state=None,
  selection='cyclic', tol=0.0001, warm_start=False)

如果我使用脚本提交工作,套索回归会自动并行运行吗?

2 个答案:

答案 0 :(得分:1)

否,它不能在并行线程上运行。相反,您可以提到n_jobs = -1(默认情况下-1使用CPU中的所有内核)

答案 1 :(得分:0)

不。我相信它不能在多个线程上运行