Comment obtenir une colonne prédite dans R

Question

Je travaille sur un modèle d'apprentissage automatique dans R en utilisant le package bayes naïf.

Voici la structure du modèle 'readFromCsv contient des multiples de lignes et de colonnes avec des valeurs valides. Je fais la prédiction, et cela me donne l'une des classes comme prédiction.

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

mais, ce que je suis en train d'atteindre est d'obtenir les détails entiers de colonne de classe prédite, disons, classe, étudiant, marques, de qualité, etc., est la liste des lignes de jeu de données, je veux obtenir tous les valeurs de la colonne prédite.

Merci pour vos suggestions.

Answer 1

Si studentDataSet_test est votre dataframe sur lequel vous voulez prédire, puis:

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