在BaggingRegressor中使用xgboost

问题描述 投票:4回答:1

我需要在xgboost中运行BaggingRegressor,我使用xgboost

import xgboost

D_train = xgboost.DMatrix(X_train, lab_train)
D_val = xgboost.DMatrix(X_train[test_index], lab_train[test_index])
D_pred =xgboost.DMatrix( X_train[test_index])
D_test = xgboost.DMatrix(X_test)
D_ttest = xgboost.DMatrix(ttest)


xgb_params = dict()
xgb_params["objective"] = "reg:linear"
xgb_params["eta"] = 0.01
xgb_params["min_child_weight"] = 6
xgb_params["subsample"] = 0.7
xgb_params["colsample_bytree"] = 0.6
xgb_params["scale_pos_weight"] = 0.8
xgb_params["silent"] = 1
xgb_params["max_depth"] = 10
xgb_params["max_delta_step"]=2
watchlist = [(D_train, 'train')]
xg = xgboost.Booster()

print('1000')
model = xgboost.train(params=xgb_params, dtrain=D_train, num_boost_round=1000, 
                      evals=watchlist, verbose_eval=1, early_stopping_rounds=20)

y_pred1 = model.predict(D_ttest)

如何使用所有相同的参数,但在BaggingRegressor

如果我做

gdr = BaggingRegressor(base_estimator= xgboost.train( params=xgb_params,
dtrain=D_train,
num_boost_round=3000,
evals=watchlist,
verbose_eval=1,
early_stopping_rounds=20))

然后开始xgboost训练,然后是代码

gdr_model = gdr
print(gdr_model)
gdr_model.fit(X_train, lab_train)
train_pred = gdr_model.predict(X_test)

print('mse from log: ', mean_squared_error(lab_train, train_pred))

train_pred = gdr_model.predict(ttest)

没有意义,或者我错了?告诉我如何解决这个问题

python scikit-learn regression xgboost
1个回答
3
投票

Xgboost有一个Sklearn包装。尝试使用以下模板!

import xgboost
from sklearn.datasets import load_boston
from xgboost.sklearn import XGBRegressor
from sklearn.ensemble import BaggingRegressor

X,y = load_boston(return_X_y=True)

reg = BaggingRegressor(base_estimator=XGBRegressor())

reg.fit(X,y)
© www.soinside.com 2019 - 2024. All rights reserved.