proper input handling and predict_proba