Predicting categories and probabilities