第二次运行时,RandomSearchCV执行未结束

时间:2019-07-18 15:40:19

标签: python python-3.x scikit-learn lightgbm

我目前正在一个项目中,我想使用lightgbm模型运行RandomSearchCV。

第一次运行我没有问题。但是,当我第二次运行单元时,执行不会结束,必须重新启动Jupyer Notebook的内核。

import pandas as pd
from sklearn.model_selection import RandomizedSearchCV
import lightgbm as lgb
import time
import warnings
warnings.filterwarnings("ignore", category=DeprecationWarning) 

#Parameters for RandomSearchCV
lgbm_random_grid = {
                'boosting_type': ['gbdt'],
                'learning_rate': [0.85,0.8,0.9]
                }




a = pd.DataFrame(data ={'col1': [-1, 1,1,1,1,-1,-1], 'col2': [-1, 1,1,1,1,-1,-1]})
b = pd.Series(data =[0,1,1,1,1,0,0])

lgbm_random = RandomizedSearchCV(estimator = LGBMClassifier(), param_distributions= lgbm_random_grid ,  n_iter = 2, cv = 2,
verbose = 2, n_jobs = -1) 


%%time
c =lgbm_random.fit(a.values,b)

详细回报

Verbose return

当我第二次尝试运行单元时:

%%time
c =lgbm_random.fit(a.values,b)

它没有结束。我尝试更改n_jobspre_dispatch参数,但没有成功。

0 个答案:

没有答案