8bb6b5a40a539ab9f1e25131c6f3add01beb6dbe,dc_integration/distribution/cacgmm.py,CACGMMTrainer,fit,#CACGMMTrainer#Any#Any#Any#Any#Any#Any#Any#Any#,45

Before Change

            ]

        if saliency is None:
            saliency = np.ones_like(initialization[..., 0, :])

        quadratic_form = np.ones_like(initialization)
        affiliation = initialization
        for iteration in range(iterations):

After Change

            affiliation /= np.einsum("...kn->...n", affiliation)[..., None, :]
            quadratic_form = np.ones(affiliation_shape, dtype=x.real.dtype)
        elif isinstance(initialization, np.ndarray):
            assert initialization.shape == affiliation_shape, (
                initialization.shape, affiliation_shape
            )
            affiliation = initialization
            quadratic_form = np.ones(affiliation_shape, dtype=x.real.dtype)
        elif isinstance(initialization, CACGMM):
            model = initialization

Instances