Add prediction type to return the mean, variance, and mode

Additionally, this change avoids unnecessary sampling if the prediction type doesn't need it.

Author: neverfox