首页 > 解决方案 > XGBoost 的 GridSearch

问题描述

我正在Analytics Vidhaya上阅读 XGBoost 的网格搜索。它在代码中有以下内容

param_test1 = {'max_depth':range(3,10,2), 'min_child_weight':range(1,6,2)}
gsearch1 = GridSearchCV(estimator = XGBClassifier( learning_rate =0.1, 
                                                   n_estimators=140, 
                                                   max_depth=5,
                                                   min_child_weight=1, 
                                                   gamma=0, 
                                                   subsample=0.8, 
                                                   colsample_bytree=0.8,
                                                   objective= 'binary:logistic', 
                                                   nthread=4, 
                                                   scale_pos_weight=1, 
                                                   seed=27), 
                        param_grid = param_test1,
                        scoring='roc_auc',
                        n_jobs=4, iid=False, 
                        cv=5)

我的问题是如果使用网格搜索来找到更好的max_depthmin_child_weight,那么为什么将这两个参数分别设置gsearch1为 5 和 1。

此外,在我自己的代码中,当我将这两个注释掉时,结果会发生变化。这是为什么?

谢谢

标签: xgboostgrid-searchgridsearchcv

解决方案


推荐阅读