vlad.py 文件源码

def fit(self, descs, preprocess=True):
        if preprocess:
            self.stdSlr = StandardScaler()
            self.stdSlr.fit(descs)
            tmp = self.stdSlr.transform(descs)
        else:
            tmp = descs
            self.stdSlr = None
        kmeans = MiniBatchKMeans(init='k-means++', n_clusters=self.num_clusters, batch_size=10000)
        kmeans.fit(tmp)
        self.centers = kmeans.cluster_centers_
        self.clusters = kmeans.labels_
        return self.centers