K-Means聚类算法python代码,和轮廓系数,CHI,CH系数
时间: 2023-12-29 17:52:08 浏览: 73
以下是K-Means聚类算法的Python代码:
```python
import numpy as np
from sklearn.cluster import KMeans
from sklearn.metrics import silhouette_score, calinski_harabasz_score, davies_bouldin_score
# 生成随机数据
X = np.random.rand(100, 2)
# 初始化KMeans模型
kmeans = KMeans(n_clusters=3, random_state=0)
# 训练模型
kmeans.fit(X)
# 预测结果
labels = kmeans.predict(X)
# 计算轮廓系数
silhouette_score = silhouette_score(X, labels)
print('轮廓系数:', silhouette_score)
# 计算CHI系数
chi_score = calinski_harabasz_score(X, labels)
print('CHI系数:', chi_score)
# 计算CH系数
ch_score = davies_bouldin_score(X, labels)
print('CH系数:', ch_score)
```
其中,`X`为输入数据,`n_clusters`为聚类簇数,`silhouette_score`为轮廓系数,`calinski_harabasz_score`为CHI系数,`davies_bouldin_score`为CH系数。
阅读全文