bow.py 文件源码

def MiniBatchKMeans(self, X, batch=10000):
        print("in fit method", X.shape, self.k)
        kmeans = MiniBatchKMeans(init='k-means++', n_clusters=self.k, batch_size=batch)
        kmeans.fit(X)
        centers = kmeans.cluster_centers_
        clusters = kmeans.labels_
        print("shape of centers is ", centers.shape)
        return centers