

def conditional_psi(self, x):
    '\n        Compute the conditional distribution over psi given observation x and omega\n        :param x:\n        :return:\n        '
    assert (x.ndim == 2)
    Omega = np.diag(self.omega)
    Sigma_cond = inv((Omega + inv(self.Sigma)))
    kappa = kappa_vec(x).sum(0)
    mu_cond = self.dot((kappa + solve(self.Sigma, self.mu)))
    return (mu_cond, Sigma_cond)

dataset/ETHPy150Open HIPS/pgmult/pgmult/distributions.py PGMultinomial.conditional_psi/VarMisuse@64/117 `Sigma_cond`->`self`