如果想对xgboost调参,可以用sklearn里的GridSearchCV吗?
2个回答
当然可以的。
from xgboost import XGBClassifier
from sklearn.model_selection import GridSearchCV
xgb_model = XGBClassifier(n_estimators=1)
parameters = {'learning_rate': [0.01, 0.02, 0.03], 'max_depth': [4, 5, 6]}
clf = GridSearchCV(xgb_model, parameters, scoring='roc_auc')
clf.fit(data, target)
上面这个例子就是对learning rate和max depth做grid search cv。
SofaSofa数据科学社区DS面试题库 DS面经