01f06390aaa65d498b2748dbc7f60f989bbc26cc,dc_integration/distribution/cwmm.py,ComplexWatsonMixtureModelParameters,predict,#ComplexWatsonMixtureModelParameters#Any#,44

Before Change

    def predict(self, Y):

        *independent, T, D = Y.shape
        assert D < 20, (D, "Sure?")

        Y = _unit_norm(

After Change

        Returns: Affiliations with shape (..., K, T).
        
        assert np.iscomplexobj(x), x.dtype
        x = x / np.maximum(
            np.linalg.norm(x, axis=-1, keepdims=True), np.finfo(x.dtype).tiny
        )
        return self._predict(x)

    def _predict(self, x):
        Predict class affiliation posteriors from given model.

Instances