def fit(self, data):
""" fit model on data """
self.data = data
kmeans = KMeans(n_clusters=self.n_clusters)
kmeans.fit(data)
self.clusterer = kmeans
logging.info('Fit has been completed')
self.data_clusters = self.clusterer.predict(data)
self.cluster_centers = self.clusterer.cluster_centers_
logging.info('Cluster calculation has been completed')
self.__clusters_separation()
logging.info('Cluster separation has been completed')
self.__cluster_avg_distances()
logging.info('Cluster avg distances has been calculated')
评论列表
文章目录