function mean_prediction_probabilities add, can help to improve the result