我目前正在一个项目中,我想使用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)
当我第二次尝试运行单元时:
%%time
c =lgbm_random.fit(a.values,b)
它没有结束。我尝试更改n_jobs
和pre_dispatch
参数,但没有成功。