k-means算法是一种常见的无监督学习算法,用于将数据聚类成不同的类别。k-means++算法是k-means算法的改进版本,旨在提高初始聚类中心选择的效率和准确性。本文将详细介绍
python
中的k-means++算法的原理、代码实现和应用。
K-means算法概述
K-means算法是一种迭代算法,每次迭代的过程为:首先随机选取K个初始聚类中心,然后将每个数据点分配到到离其最近的初始聚类中心所在的类别中,接着重新计算所有聚类的中心,并更新聚类中心。重复上述过程直至满足收敛条件。
K-means的算法流程:
随机从数据中选取K个数据点作为初始聚类中心。
将数据点分配到距离其最近的聚类中心所在的聚类中。
重新计算每个聚类的中心。
重复2-3,直到满足收敛条件(聚类中心不再变化、达到最大迭代次数等)。
K-means++算法步骤
K-means++算法是K-means算法的改进版本,主要在初始聚类中心的选择上做了优化。K-means++算法的初始聚类中心选择步骤如下:
随机选择一个数据点作为第一个聚类中心。
对于每个数据点,计算它与最近的聚类中心的距离D(x)。
随机选择一个数据点作为下一个聚类中心,要保证这个点与已有的聚类中心的距离越大,被选中的概率越大:
a. 计算每个数据点与已有聚类中心的最近距离D(x)^2。
b. 计算所有D(x)^2的和Sum(D(x)^2)。
c. 按照每个数据点与已有聚类中心的最近距离所占比例为其赋权重,概率为D(x)^2/Sum(D(x)^2)。
d. 从按照上述概率进行抽样,选择一个数据点作为下一个聚类中心。
重复步骤3,直到选取K个聚类中心。
Python实现K-means++算法
下面,我们将通过Python实现K-means++算法。
立即学习
“
Python免费学习笔记(深入)
”;
首先,导入必要的库:
接着,我们生成一组用于聚类的数据:
Python 3.14.3
微软官方的 Python 扩展,是 VS Code 安装量最高的扩展(209M+)。集成 IntelliSense(通过 Pylance)、调试(通过 Python Debugger)、代码检查、格式化、重构和单元测试等功能。支持 Jupyter Notebook、虚拟环境管理和多 Python 版本切换。
下载
然后,我们通过sklearn的KMeans模块来训练K-means++模型:
最后,我们将聚类结果可视化:
K-means++算法应用场景
K-means算法适用于没有标签信息的数据聚类问题。K-means++算法相比于K-means算法,为保证初始聚类中心的合理性和独特性,更加适用于数据较多或者数据分布比较分散的情况下。
K-means++算法可以用于数据挖掘、图像处理、自然语言处理等领域。可以利用聚类算法找出相似性较高的样本,对于大数据的可视化也有很好的应用。
总之,K-means++算法在数据挖掘、聚类分析、图像识别、自然语言处理等领域都有很好的应用前景。
import numpy as np
import matplotlib.pyplot as plt
from sklearn.datasets import make_blobs
from sklearn.cluster import KMeansn_samples = 1500
random_state = 170
X, y = make_blobs(n_samples=n_samples, random_state=random_state)kmeans = KMeans(init="k-means++", n_clusters=3, n_init=10)
kmeans.fit(X)plt.figure(figsize=(12, 12))
h = 0.02
x_min, x_max = X[:, 0].min() - 1, X[:, 0].max() + 1
y_min, y_max = X[:, 1].min() - 1, X[:, 1].max() + 1
xx, yy = np.meshgrid(np.arange(x_min, x_max, h), np.arange(y_min, y_max, h))
Z = kmeans.predict(np.c_[xx.ravel(), yy.ravel()])
Z = Z.reshape(xx.shape)
plt.imshow(Z, interpolation="nearest",
extent=(xx.min(), xx.max(), yy.min(), yy.max()),
cmap=plt.cm.Pastel1, aspect="auto", origin="lower")
plt.scatter(X[:, 0], X[:, 1], s=30, c=kmeans.labels_, cmap=plt.cm.Paired)
plt.scatter(kmeans.cluster_centers_[:, 0], kmeans.cluster_centers_[:, 1],
marker="^", s=100, linewidths=3,
color='black', zorder=10)
plt.title("K-means++ clustering")
plt.xlim(x_min, x_max)
plt.ylim(y_min, y_max)
plt.show()