sklearn.cluster.kmeans导入数据集代码示例
时间: 2023-06-09 21:02:59 浏览: 103
KMeans聚类算法+代码
5星 · 资源好评率100%
好的,您可以使用以下代码示例导入数据集并进行 k-means 聚类:
```python
from sklearn.datasets import load_iris
from sklearn.cluster import KMeans
# 加载 iris 数据集
iris = load_iris()
# 获取特征数据
X = iris.data
# 定义 k-means 聚类模型
kmeans = KMeans(n_clusters=3)
# 将数据集聚类
kmeans.fit(X)
# 输出数据集聚类结果
print(kmeans.labels_)
```
这段代码导入了 sklearn 库中的 iris 数据集,然后使用 k-means 算法将数据集聚类成 3 类,并输出每个样本所属的类别。
阅读全文