我想使用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)
如果我使用脚本提交工作,套索回归会自动并行运行吗?
答案 0 :(得分:1)
否,它不能在并行线程上运行。相反,您可以提到n_jobs = -1(默认情况下-1使用CPU中的所有内核)
答案 1 :(得分:0)
不。我相信它不能在多个线程上运行