使用sklearn做聚类分析
时间: 2023-07-18 15:11:13 浏览: 116
大数据分析技术-Sklearn操作与聚类分析模型构建与评价.doc
5星 · 资源好评率100%
好的,我可以为您解答使用sklearn做聚类分析的问题。Sklearn是一个流行的Python机器学习库,它提供了多种聚类算法,包括K均值聚类、层次聚类、DBSCAN等。下面是一个使用K均值聚类算法的示例代码:
```
from sklearn.cluster import KMeans
import numpy as np
# 创建数据集
X = np.array([[1, 2], [1, 4], [1, 0],
[4, 2], [4, 4], [4, 0]])
# 定义K均值聚类模型
kmeans = KMeans(n_clusters=2, random_state=0)
# 训练模型并进行聚类
kmeans.fit(X)
# 输出聚类结果
print(kmeans.labels_)
```
在这个示例中,我们创建了一个2维的数据集X,其中包含6个样本。我们定义了一个K均值聚类模型,将数据集分为2个簇。然后我们使用fit()方法训练模型,并使用labels_属性输出聚类结果。
您可以根据自己的数据集和需求修改代码,并使用其他聚类算法进行实验。希望这个示例能够帮助到您。
阅读全文