FIX: use predict_log_proba when available