def psi(self, data):
        # TODO: Fix this hack
        if "z" in data:
            z = data["z"]
        elif "states" in data:
            z = data["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/original