如何在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