【代码模版】轮廓系数确定聚类类数及可视化展示

import matplotlib
import matplotlib.pyplot as plt
%matplotlib inline
plt.rcParams['axes.labelsize'] = 14
plt.rcParams['xtick.labelsize'] = 12
plt.rcParams['ytick.labelsize'] = 12
from sklearn.cluster import KMeans
many_models = [KMeans(n_clusters = n).fit(X) for n in range(a, b)] # 指定多个n值来训练多个模型
from sklearn.metrics import silhouette_score
silhouette_scores = [silhouette_score(X,model.labels_) for model in many_models]  # 得到多个n值对应的轮廓系数
plt.figure(figsize=(8,4))
plt.plot(range(a, b),silhouette_scores,'bo-')
plt.show()