DataWhale组队sklearn学习 -- K-Means

k-means

1. 用于无监督学习
2. 用于分类
3. 思想：我们假设相似的事物之间的距离也会比较近，即：“物以类聚，人以群分”的思想。因此对于给定的样本集， 按照样本之间的距离，将样本集划分为k个簇。然后让簇内尽量紧密，簇间尽量距离大，从而实现分类。
4. 特点：
   1. 优点
      1. 思路比较简单
      2. 实现简单
      3. 聚类效果不错
      4. 可解释性强
   2. 缺点
      1. 对于噪点很敏感。如果在远处有一个早点，那么簇中心的位置就可能有很大的偏移
      2. k 值很难确定
      3. 如果两个簇间的距离比较近，此时效果也不好
      4. 初始值对结果影响很大，因此容易出现每次聚类的结果均不一样
5. 过程：
   1. 随机选择 k 个样本作为簇心
   2. 计算各个样本到簇中心的距离，将最小的距离的作为此类别
   3. 迭代计算，直至类别(簇心)不再发生变化
   4. 输出类别

import matplotlib.pyplot as plt
from sklearn.datasets import make_blobs # 选择内置的生成聚类的数据集
from sklearn.cluster import KMeans
# make_blobs：生成聚类的数据集
# n_samples：生成的样本点个数，n_features：样本特征数，centers：样本中心数
# cluster_std：聚类标准差，shuffle：是否打乱数据，random_state：随机种子
# 300数据，3个样本特征，四个中心点(k=4),
x, y = make_blobs(n_samples=300, n_features=3,centers=4, cluster_std=0.5,shuffle=True, random_state=0)
plt.scatter(x[:, 0], x[:, 1], c='green', marker='o', edgecolors='black', s=50)
plt.show()



x, y = make_blobs(n_samples=300, n_features=3,centers=10, cluster_std=0.5,shuffle=True, random_state=0)
plt.scatter(x[:, 0], x[:, 1], c='green', marker='o', edgecolors='black', s=50)
plt.show()


X, y = make_blobs(n_samples=150, n_features=2,centers=3, cluster_std=0.5,shuffle=True, random_state=0)
# 散点图
# c：点的颜色，marker：点的形状，edgecolor：点边缘的形状，s：点的大小
plt.scatter(X[:, 0], X[:, 1],c='white', marker='o',edgecolor='black', s=50)
plt.show()

 

 

 由上可以看出，即使是手动生成的数据，当k值过大的时候，效果有时候也会不尽如人意。 下面我使用k为3的数据集做训练，分类

# 定义模型
# n_clusters：要形成的簇数，即k均值的k，init：初始化方式，tot：Frobenius 范数收敛的阈值
model = KMeans(n_clusters=3, init='random',n_init=10, max_iter=300, tol=1e-04, random_state=0)
# 训练加预测
y_pred = model.fit_predict(X)
# 画出预测的三个簇类
plt.scatter(
    X[y_pred == 0, 0], X[y_pred == 0, 1],
    s=50, c='lightgreen',
    marker='s', edgecolor='black',
    label='cluster 1'
)

plt.scatter(
    X[y_pred == 1, 0], X[y_pred == 1, 1],
    s=50, c='orange',
    marker='o', edgecolor='black',
    label='cluster 2'
)

plt.scatter(
    X[y_pred == 2, 0], X[y_pred == 2, 1],
    s=50, c='lightblue',
    marker='v', edgecolor='black',
    label='cluster 3'
)

# 画出聚类中心
plt.scatter(
    model.cluster_centers_[:, 0], model.cluster_centers_[:, 1],
    s=250, marker='*',
    c='red', edgecolor='black',
    label='centroids'
)
plt.legend(scatterpoints=1)
plt.grid()
plt.show()

 

 

# 计算inertia随着k变化的情况
distortions = []
for i in range(1, 10):
    model = KMeans(
        n_clusters=i, init='random',
        n_init=10, max_iter=300,
        tol=1e-04, random_state=0
    )
    model.fit(X)
    distortions.append(model.inertia_)
# 画图可以看出k越大inertia越小，追求k越大对应用无益处
plt.plot(range(1, 10), distortions, marker='o')
plt.xlabel('Number of clusters')
plt.ylabel('Distortion')
plt.show()