输出随机森林的预测值作为测试数据集中的新列

Question

我运行训练数据集上的运行dom 森林（父数据集的 60%）。

我使用 R 中的 "predict" 函数来评估模型在测试数据集（父数据集的 40%）上的表现

我用过

 require(randomForest)

 trained_model <- randomForest(y~.,data = training,
                                   importance=TRUE, 
                                   keep.forest=TRUE)

 result <- predict(trained_model, type = response, newdata=testing[-17])

删除了第 17 列，因为它是 "y"

问题是 "result" 是一个数组，这有办法将这个预测值作为新列添加回测试数据集中

类似于

Id     x1     x2 ......   Xn    y     predicted
1      0.1    0.12        23    no    no

Answer 1

这样做：

testing[ ,(ncol(testing)+1)] <- result

将结果添加回测试数据集时，会有一列而不是数组。

输出随机森林的预测值作为测试数据集中的新列

output the predicted value of random forest as a new column in the test dataset

r

random-forest