Ошибка при вызове model.labels в KMeans

Question

Я использую этот код

import pandas as np
import numpy as np

from sklearn import cluster
from sklearn.cluster import KMeans

model = cluster.KMeans(n_clusters=4, random_state=10)

Затем я передаю это через фрейм данных, над которым я работаю, и включает столбцы возраста и дохода, то есть кластеры, над которыми я работаю,

model.fit(df[['income', 'age']]

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

df['cluster'] = model.labels_df.head()

И это ошибкакод, который я получаю:

AttributeError: 'KMeans' object has no attribute 'labels_df'

Есть предложения?

Answer 1

Атрибут для доступа к меткам модели: model.labels_

Использование:

df['cluster'] = model.labels_

При вводе model.labels_df.head() вы запрашиваете заголовокmodel.labels_df которого не существует.

Я полагаю, что вы его опечатали, и вам нужно:

df['cluster'] = model.labels_
df.head()