Data Science Desktop Survival Guide
by Graham Williams |
|||||
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)
|