xgb的目标函数:
选择参数:
可选择的函数:
在xgb中训练:
# 在xgb中训练
import xgboost
# DMatrix 读取数据
dm_train = xgboost.DMatrix(x_train,y_train)
dm_test = xgboost.DMatrix(x_test,y_test)
# 选择参数
param = {'silent':False,'objective':'reg:linear','eta':0.1}
num_round = 180
# 模型训练
bst = xgboost.train(param,dm_train,num_round)
y_pre = bst.predict(dm_test)
from sklearn.metrics import r2_score
r2_score(y_test,y_pre)
mse(y_test,y_pre)
在xgb中的训练效果是要优于sklearn的
最优化: