Sklearn MiniBatchKMeans дает непонятные результаты для атрибута label_ - PullRequest
0 голосов
/ 23 сентября 2019

Я использую sklearn.cluster.MiniBatchKMeans для обучения модели ML.Мне нужно получить идентификаторы кластеров кластеров, и я попытался с приведенным ниже кодом.(Здесь модель - модель кластеризации MiniBatchKmeans)

print("Cluster IDs: ", np.unique(model.labels_))
print("Number of Clusters: ", model.n_clusters)

Я получил следующий результат.

Cluster IDs:  [0]
Number of Clusters:  2

Согласно этому результату он показывает, что существует только1 идентификатор кластера для данного набора данных, и все же есть 2 кластера.Я обнаружил, что все новые данные (2-й этап обучения), которые я использовал для обучения модели с использованием функции partal_fit , были назначены одному кластеру.Кажется, что центры кластеров обновляются должным образом, но функция transform дает неверные результаты для вектора расстояния.В чем причина?

...