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

在 KMeans 中调用 model.labels 时出错

Error when calling model.labels in KMeans

python

k-means

pandas

scikit-learn