学习笔记 Day 50（XGBoost）

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的

最优化：