Predikce po natrénování modelu

Supervised Learning in R: Regression

Nina Zumel and John Mount

Win-Vector LLC

Predikce na trénovacích datech

cricket$prediction <- predict(cmodel)

predict() ve výchozím nastavení vrací predikce na trénovacích datech

Zobrazení predikcí

ggplot(cricket, aes(x = prediction, y = temperature)) + 
     geom_point() +
     geom_abline(color = "darkblue") + 
     ggtitle("temperature vs. linear model prediction")

Predikce na nových datech

newchirps <- data.frame(chirps_per_sec = 16.5)
newchirps$prediction <- predict(cmodel, newdata = newchirps)
newchirps

   chirps_per_sec     pred
 1           16.5 79.53537

Pojďme procvičovat!

Supervised Learning in R: Regression