7a230980914c8a0223f0e04a54c9e6f240358e05,tslearn/clustering.py,KShape,_fit_one_init,#KShape#Any#Any#,741

Before Change

        if hasattr(self.init, "__array__"):
            self.cluster_centers_ = self.init.copy()
        else:
            self.cluster_centers_ = rs.randn(self.n_clusters, sz, d)
        self._norms_centroids = numpy.linalg.norm(self.cluster_centers_, axis=(1, 2))
        if hasattr(self.init, "__array__"):
            self._assign(X)
        else:

After Change

            self.cluster_centers_ = self.init.copy()
        elif self.init == "random":
            indices = rs.choice(X.shape[0], self.n_clusters)
            self.cluster_centers_ = X[indices].copy()
        else:
            raise ValueError("Value %r for parameter "init" is invalid" % self.init)
        self._norms_centroids = numpy.linalg.norm(self.cluster_centers_, axis=(1, 2))
        self._assign(X)

Instances