fix predict_proba value in testing function