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