0
点赞
收藏
分享

微信扫一扫

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

洲行 2022-02-08 阅读 183

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
举报

相关推荐

#R语言实战2

0 条评论