在KMeans中调用model.labels时出错

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()