R语言实战应用精讲50篇（三十五）-R语言实现xgboost回归（附R语言代码）

Posted 2022-02-18 文宇肃然

xgboost回归

采用xgboost包构建xgboost模型有一个比较棘手问题在于构建xgb.DMatrix数据，同时为了应用提前终止策略需要构建watchlist，以下代码从数据准备部分开始对上面两部分内容均有覆盖，大家可以参考使用。

# 数据准备colnames(boston)# 训练集dvfunc <- dummyVars(~., data = data_train[, 1:13], fullRank = T)data_trainx <- predict(dvfunc, newdata = data_train[, 1:13])data_trainy <- data_train$medv# 验证集data_validx <- predict(dvfunc, newdata = data_valid[, 1:13])data_validy <- data_valid$medv# 测试集data_testx <- predict(dvfunc, newdata = data_test[, 1:13])data_testy <- data_test$medv# 构建xgb.DMatrix格式数据dtrain <- xgb.DMatrix(data = data_trainx, label = data_trainy)dvalid <- xgb.DMatrix(data = data_validx, label = data_validy)dtest <- xgb.DMatrix(data = data_tes