

def psi(self, data):
    if ('z' in data):
        z = data['z']
    elif ('states' in data):
        z = z['states'].stateseq
    else:
        raise Exception('Could not find latent states!')
    psi = (z.dot(self.C.T) + self.mu[None, :])
    return psi

dataset/ETHPy150Open HIPS/pgmult/pgmult/distributions.py PGMultinomialRegression.psi/VarMisuse@40/49 `data`->`z`