now can use predict_probs