目前被分配的聚类中心的样本 指数 (1,2,…, )
聚类中心 ( )
已经被分配聚类中心的样本
import numpy as np # 导入numpy库
import matplotlib.pyplot as plt # 导入matplotlib.pylab库
from sklearn.cluster import KMeans # 导入sklearn.cluster库
estimator = KMeans(n_clusters=4)
主要参数:
n_clusters : 隐聚类中心数目, 缺省为8
init : 初始化方法 {‘k-means++’, ‘random‘, ndarray}, 缺省’k-means++’ (智能选择聚类中心)
n_init : k-means算法在以不同的聚类中心组合运行的次数,缺省10
max_iter : k-means算法每次运行时,最大迭代次数,缺省300
tol : 容忍度, 可选, 没两次代价的差值达到10-4就停止了。缺省le-4, 达到此值即停止
precompute_distances : 提前计算距离{‘auto’, True, False} ‘auto’: 若n_samples * n_clusters > 12 million 不进行
random_state: 缺省None; 若int, 随机数产生器seed, 若RandomStates实例, 随机数产生器, 若None, np.random
缺点:结果高度依赖于初始值的设置
estimate.predict(X) #预测数据集X中每个样本所属的聚类中心索引
estimate.fit_predict(X) #计算聚类并预测每个样本的中心:[索引1, 索引2…]
estimate.transform(X) #将样本数据集X转换到(聚类-距离)空间shape(m,k)
estimate.fit_transform(X) # 计算聚类并将X转换到(聚类-距离)空间: shape(m,k)
estimate.cluster_centers_ #聚类中心坐标: shape(k, n_features)
estimate.labels_ #每个样本的所属中心标签索引,同predict(X),shape(m,)
estimate.inertia_ # 所有样本与其最近中心距离的平方和
estimate.score(X) # k-mean算法目标值的相反值, 即inertia_相反值
def showCluster(X, k, centroids, clusterAssment):
m, dim = X.shape #样本数m,维度dim
# 画所有样本
mark = [‘or’, ‘ob’, ‘og’, ‘ok’, ‘^r’, ‘+r’, ‘sr’, ‘dr’, ‘
#画肘部法则曲线
k=[]
inertia=[]
for i in range(data.shape[0]-1):
estimator = KMeans(n_clusters=i+1)
res = estimator.fit_predict(data)
k.append(i+1)
inertia.append(estimator.inertia_); #print(int(inertia))
plt.figure('肘部法则')
plt.title('肘部曲线')
plt.plot(k,inertia)
plt.scatter(k,inertia,c='r')
#画标注
for i in range(data.shape[0]-1):
plt.annotate(str(k[i]), xy=(k[i], inertia[i]), xytext=(+0, +4),
textcoords='offset points', fontsize=10)
plt.show()