ENH: refactor predict and predict_proba + better doc