import itertools from sklearn.metrics import silhouette_score from sklearn.cluster import KMeans k_list = range(2,10,1) for k in itertools.product(k_list): km = KMeans(n_clusters=k) labels = km.fit(data).labels_报错
时间: 2023-12-27 17:03:26 浏览: 109
Python itertools.product方法代码实例
上述代码还是存在问题,应该修改为如下形式:
```python
import itertools
from sklearn.metrics import silhouette_score
from sklearn.cluster import KMeans
k_list = range(2, 10, 1)
for k in k_list:
km = KMeans(n_clusters=k)
labels = km.fit_predict(data)
```
在进行KMeans聚类时,应该使用`fit_predict`方法来直接获取样本所属的簇标签,而不是使用`fit`方法再调用`labels_`属性来获取簇标签。
阅读全文