14.7 Predict Probability

20200607 To extract the probability of it raining tomorrow from the prediction made by the model. For a rpart model we can obtain the probability of it raining tomorrow using stats::predict() with type="prob".

ds[tr, vars] %>% predict(model, newdata=., type="prob") %>% '['(,2) -> pr_tr
ds[tu, vars] %>% predict(model, newdata=., type="prob") %>% '['(,2) -> pr_tu
ds[te, vars] %>% predict(model, newdata=., type="prob") %>% '['(,2) -> pr_te
head(pr_tr)
##         1         2         3         4         5         6 
## 0.7639485 0.1409460 0.1409460 0.1409460 0.1409460 0.1409460


Your donation will support ongoing development and give you access to the PDF version of this book. Desktop Survival Guides include Data Science, GNU/Linux, and MLHub. Books available on Amazon include Data Mining with Rattle and Essentials of Data Science. Popular open source software includes rattle, wajig, and mlhub. Hosted by Togaware, a pioneer of free and open source software since 1984.
Copyright © 1995-2021 Graham.Williams@togaware.com Creative Commons Attribution-ShareAlike 4.0.