FIX: predict_log_proba (again)