e93c1a7e2d0a730c7de590b8a18eae09f313f161,dc_integration/distribution/gmm.py,GMM,predict,#GMM#Any#,12

Before Change

    gaussian: Gaussian

    def predict(self, x):
        raise NotImplementedError


class GMMTrainer:
    def __init__(self, eps=1e-10):

After Change

        affiliation = np.zeros(affiliation_shape)
        affiliation += np.log(self.weight)[..., :, None]
        affiliation += self.gaussian.log_pdf(x)
        affiliation = np.exp(affiliation)
        denominator = np.maximum(
            np.einsum("...kn->...n", affiliation)[..., None, :],
            np.finfo(x.dtype).tiny,
        )
        affiliation /= denominator
        return affiliation


class GMMTrainer:
    def __init__(self, eps=1e-10):

Instances