xgboost GridSearchCV花费的时间太长或无法进行下一步

时间:2019-04-15 01:44:04

标签: machine-learning xgboost gridsearchcv

奇怪

%%time
xgb = xgb.XGBRegressor(n_estimators=500, learning_rate=0.07, gamma=0, subsample=0.75, colsample_bytree=1,
                            max_depth=7, tree_method='gpu_exact')

此代码大约需要Wall时间:866毫秒。

但是当我执行gridsearchCV时,它不会进入下一步
即使我只给出了一个参数

    %%time
    xgb = XGBClassifier(tree_method='gpu_exact',verbose_eval=True, silence=False)
    kfold = StratifiedKFold(n_splits=10, random_state=0)
    xgb_param_grid = {
                    'learning_rate': [0.08,0.09],
                    'random_state': [0],
                    'max_depth': [8,9],
                    'n_estimators': [400,500]
    }
    xgbGrid = gsRFC = GridSearchCV(xgb,param_grid = xgb_param_grid, cv=5, scoring="neg_mean_squared_error", n_jobs= 10, verbose = 1)
    xgbGrid.fit(X,y)
    xgb_best = xgbGrid.best_estimator_

for my understanding, this should not take that long.<br/>
it d

不进行下一步,我不确定这是否可行
它以

停止
  

为8个候选人中的每一个进行5折,总计40个合身   [Parallel(n_jobs = 10)]:使用后端LokyBackend并发10个   工人。

数据集大小为(15035,22) 难道我做错了什么?

0 个答案:

没有答案