本关任务::你需要调用 sklearn 中的K-means模型,对红酒数据进行聚类。
时间: 2024-02-18 19:01:48 浏览: 102
实现K-means算法,并在红酒数据集上运行
好的,我知道了。现在我已经加载了红酒数据集,并准备使用 sklearn 中的 KMeans 模型进行聚类。下面是代码实现:
```python
from sklearn.cluster import KMeans
from sklearn.datasets import load_wine
# 加载红酒数据集
wine = load_wine()
# 使用 KMeans 进行聚类
kmeans = KMeans(n_clusters=3)
kmeans.fit(wine.data)
# 输出聚类结果
print(kmeans.labels_)
```
这里我设置了聚类数为 3,你可以根据具体需求进行修改。聚类结果会被保存在 `kmeans.labels_` 中,可以输出查看。
阅读全文