predict can handle either the index of the case of interest, or a single probability representing a point prediction (p_hat).