Wie bekomme ich vorhergesagte Spalte in R

Question

Ich arbeite an einem maschinellen Lernmodell in R mit naiven Bayes-Paket.

Im Folgenden sind die Modellrahmen 'readFromCsv enthält ein Vielfaches von Zeilen und Spalten mit gültigen Werten.

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

Ich mache die Vorhersage, und das gibt mir eine der Klassen als die Vorhersage.

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

aber, was ich versuche, die gesamte Spalte Details aus vorhergesagten Klasse erhalten zu erreichen ist, sagen wir, klasse, student, Noten, Note etc., ist die Liste der Zeilen in Datensatzes, ich will alle bekommen die Werte aus der vorhergesagten Spalte.

Vielen Dank für Anregungen.

Answer 1

Wenn studentDataSet_test ist Ihre Datenrahmen, über die Sie vorhersagen wollen, dann:

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