

def augment_data(self, augmented_data):
    '\n        Augment the data with auxiliary variables\n        :param augmented_data:\n        :return:\n        '
    x = augmented_data['x']
    (T, K) = x.shape
    assert (K == self.K)
    augmented_data['kappa'] = kappa_vec(x)
    augmented_data['omega'] = np.ones((T, (K - 1)))
    self.resample_omega([augmented_data])
    return x

dataset/ETHPy150Open HIPS/pgmult/pgmult/distributions.py PGMultinomialRegression.augment_data/VarMisuse@80/141 `augmented_data`->`x`