在R的XG-Boost中使用predict（）时发生错误

Question

我一直在运行其他算法，根据其他几个参数来预测Facebook帖子的效果。我要尝试的最后一种方法是XG-Boost。

即使重新检查我的代码和软件包文档后，我仍然不断出错。我的火车和测试数据都已清除，所有因素都已转换为1和0的列。

//处理测试和训练数据

temp.treat <- prepare(treatplan,temp, varRestriction = newvars)
test.treat <- prepare(treatplan,test, varRestriction = newvars) 

//训练模型

cv <- xgb.cv(data = as.matrix(temp.treat),
         label = temp$Reach,
         objective = "reg:linear",
         nrounds = 400, nfold = 5, eta = 0.3, depth = 6) 

//获得预测

test$pred <- predict(cv, as.matrix(test.treat))

数据训练时没有引发错误，但是运行预测命令的那一刻，我得到了错误-

UseMethod（“ predict”）中的错误：没有适用于“预测”的适用方法应用于类“ xgb.cv.synchronous”的对象]

有人可以告诉我我在做什么错吗？

Answer 1

您应该使用该功能包中的xgboost()。

xgb.cv()只能帮助您获取$ evaluation表以获取有关]的信息>