FIX: predict_log_proba