Как обеспечить центроиды кластеров в k означает, что алгоритм не переключается каждый раз?

У меня есть CSV-файл, который выглядит как показано ниже

date                       mse                                                  
2018-02-11                 14.34
2018-02-12                 7.24
2018-02-13                 244.5
2018-02-14                 3.5
2018-02-16                 12.67
2018-02-21                 45.66
2018-02-22                 15.33
2018-02-24                 98.44
2018-02-26                 23.55
2018-02-27                 45.12
2018-02-28                 78.44
2018-03-01                 34.11
2018-03-05                 23.33
2018-03-06                 127.45
...                        ...   
...                        ...

Теперь я пытаюсь применить k означает к значениям mse, чтобы получить 2clusters, который дает мне 2centroids по одному для каждого. Теперь мне дано значение mse, и мне нужно найти, какой из двух centroids ближе к заданному значению mse. сделай что-нибудь вроде этого

from sklearn.cluster import KMeans
import pandas as pd

centroid_list = []
given_mse = 7.382409087
kmeans = KMeans(n_clusters=2)
df = pd.read_csv("data.csv", parse_dates=["date"])
kmeans.fit_predict(df[['mse']])
centroid_list.append(kmeans.cluster_centers_.ravel())
#print(centroids_list)  # array([  153.27996598, 19810.6925875 ]
for i in centroids_list:
t1 = abs(given_mse - i[0])
t2 = abs(given_mse - i[1])
if t1 < t2:
   result.append("label 1")
   else:
   result.append("label 2")
print(result)  # ['label1']

Теперь, как вы можете видеть, я получил два значения centroid: 153.27996598 и 19810.6925875, назначенные каждому кластеру.

Проблема в том, что он часто переключает значения [(x, y) или (y, x)], когда вы запускаете программу, из-за чего я получаю конечный результат либо как label1, либо как иногда label2.

Есть идеи, как это исправить. Есть ли какой-нибудь метод sckit-learn, чтобы предотвратить это переключение?

Вы можете сохранить модель и перезагрузить ее, а не переучивать.

piman314 02.05.2018 12:20

@ncfirth Мне очень жаль, но я не совсем уверен, что понимаю. Не могли бы вы уточнить?

Souvik Ray 02.05.2018 12:27

Создайте отдельный скрипт, в котором вы обучаете модель и сохраняете ее, пример можно увидеть здесь. Затем каждый раз, когда вы используете свою модель, просто загружайте сохраненную, а не переучивайтесь.

piman314 02.05.2018 12:32

Используйте параметр random_state в KMeans

Vivek Kumar 02.05.2018 12:47

@VivekKumar спасибо! Работает!

Souvik Ray 02.05.2018 15:19
Стоит ли изучать PHP в 2023-2024 годах?
Стоит ли изучать PHP в 2023-2024 годах?
Привет всем, сегодня я хочу высказать свои соображения по поводу вопроса, который я уже много раз получал в своем сообществе: "Стоит ли изучать PHP в...
Поведение ключевого слова "this" в стрелочной функции в сравнении с нормальной функцией
Поведение ключевого слова "this" в стрелочной функции в сравнении с нормальной функцией
В JavaScript одним из самых запутанных понятий является поведение ключевого слова "this" в стрелочной и обычной функциях.
Приемы CSS-макетирования - floats и Flexbox
Приемы CSS-макетирования - floats и Flexbox
Здравствуйте, друзья-студенты! Готовы совершенствовать свои навыки веб-дизайна? Сегодня в нашем путешествии мы рассмотрим приемы CSS-верстки - в...
Тестирование функциональных ngrx-эффектов в Angular 16 с помощью Jest
В системе управления состояниями ngrx, совместимой с Angular 16, появились функциональные эффекты. Это здорово и делает код определенно легче для...
Концепция локализации и ее применение в приложениях React ⚡️
Концепция локализации и ее применение в приложениях React ⚡️
Локализация - это процесс адаптации приложения к различным языкам и культурным требованиям. Это позволяет пользователям получить опыт, соответствующий...
Пользовательский скаляр GraphQL
Пользовательский скаляр GraphQL
Листовые узлы системы типов GraphQL называются скалярами. Достигнув скалярного типа, невозможно спуститься дальше по иерархии типов. Скалярный тип...
0
5
45
1
Перейти к ответу Данный вопрос помечен как решенный

Ответы 1

Ответ принят как подходящий

Как упоминал @Vivek Kumar, мне нужно было передать дополнительный параметр random_state при установке средних k. Значение для random_state может быть любым целым числом.

kmeans = KMeans(n_clusters=2, random_state=1)

Другие вопросы по теме