predict function

Model predictions