En utilisant bayésiens naïfs pour prédire de nouvelles valeurs

Question

j'ai un dataframe qui ressemble à ceci

weather <- c("good", "good", "good", "bad", "bad", "good") 
temp <- c("high", "low", "low", "high", "low", "low") 
golf <- c("yes", "no", "yes", "no", "yes" , "no") 
df <- data.frame(weather, temp, golf) 

Ce que je voudrais faire est maintenant en utilisant la méthode bayésiens naïfs pour obtenir la probabilité de ce nouvel ensemble de données

df_new <- data.frame(weather = "good", temp = "low") 

donc j'essaie

library(e1071) 
model <- naiveBayes(golf ~.,data=df) 
predict(model, df_new) 

Mais cela me donne: Une idée comment je peux transformer cela en une probabilité?

Answer 1

probabilités sont renvoyés si vous utilisez type = "raw"

predict(model, df_new, type = "raw") 
no yes 
[1,] 0.5 0.5 

predict(model, df_new, type = "class") 
[1] no 
Levels: no yes