def conditional_mean(self, augmented_data):
        cm = augmented_data["kappa"] / augmented_data["omega"]
        cm[~np.isfinite(cm)] = 0
        cm -= self.mu[None,:]
        return cm
Compute the conditional mean \psi given \omega
dataset/ETHPy150Open HIPS/pgmult/pgmult/distributions.py/PGMultinomialRegression.conditional_mean