如何在R中获得预测列

Question

我正在使用朴素贝叶斯包在 R 中研究机器学习模型。

以下是模型框架' readFromCsv 包含多个具有有效值的行和列。

model <- naiveBayes(student~ ., data = studentDataSet)

我做预测，这给了我一个类作为预测。

predict(model, studentDataSet[1:10,], type = "class")

但是，我想要实现的是从预测的班级中获取整个列的详细信息，例如班级、学生、分数、成绩等，是数据集中的行列表，我想从预测中获取所有值柱子。

感谢您的任何建议。

Answer 1

如果 studentDataSet_test 是您要预测的数据框，那么：

studentDataSet_test$prediction <- predict(model, studentDataSet_test, type = "class")
print(studentDataSet_test)
# This will print all details with the prediction