news 2026/6/15 19:47:47

均值漂移聚类Mean Shift Cluster用于数据聚类,代码注释详细,适合新手学习~

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
均值漂移聚类Mean Shift Cluster用于数据聚类,代码注释详细,适合新手学习~

均值漂移聚类Mean Shift Cluster用于数据聚类,代码注释详细,适合新手学习~

均值漂移聚类这名字听起来挺学术,但说白了就是个"随大流"的算法。它的核心思想特有意思:想象你站在一片丘陵地带,蒙着眼睛往四周扔石子,听到石子落地声最大的方向就迈一步,重复这操作直到爬上山顶——这就是密度最高的地方。搞机器学习的人把这过程叫做"密度梯度上升",咱们普通人理解成"哪人多往哪挤"就行了。

先看个实际场景:假设我要把淘宝买家的消费行为数据分成不同群体。咱们不整那些虚的,直接上代码:

from sklearn.datasets import make_blobs import matplotlib.pyplot as plt # 生成3簇数据,特意加了些重叠区域更真实 X, _ = make_blobs(n_samples=500, centers=3, cluster_std=1.8, random_state=11) # 可视化原始数据分布 plt.scatter(X[:,0], X[:,1], s=15, edgecolor='k') plt.title('原始数据分布') plt.show()

!散点图显示三个有部分重叠的簇

这段代码的关键在makeblobs,它像捏面团似的给我们生成测试数据。clusterstd参数控制着数据点的分散程度,调大这个值会让各簇之间产生更多重叠,更接近真实场景中的数据分布。

接下来上主菜:

from sklearn.cluster import MeanShift # 创建模型时要注意带宽选择,相当于决定搜索范围的大小 # 这里让算法自动估算,实际项目需要交叉验证 ms = MeanShift(bandwidth=2.5, bin_seeding=True) ms.fit(X) # 提取聚类结果 labels = ms.labels_ cluster_centers = ms.cluster_centers_ # 可视化魔法 plt.scatter(X[:,0], X[:,1], c=labels, s=15, edgecolor='k') plt.scatter(cluster_centers[:,0], cluster_centers[:,1], c='red', s=250, marker='X', edgecolor='w') plt.title('聚类结果(红叉为簇中心)') plt.show()

!显示三个被正确划分的簇,中心标记明显

重点说下bandwidth参数,这相当于算法中的"社交距离"。设置太大容易把不同群体合并,太小又会产生过多小群体。新手可以试着把这个值从1改到5,观察聚类结果的变化,比看论文管用多了。

这个算法有个特点,不用预先指定簇的数量。它像扫地机器人似的在数据空间里游走,自动发现高密度区域。但代价就是计算量较大,当数据量超过十万级时可能得换其他方法。

遇到异常值怎么办?咱们举个极端例子:

# 故意添加离群点 import numpy as np outliers = np.array([[15, 15], [-10, -5], [20, -8]]) X_with_outliers = np.vstack([X, outliers]) # 重新训练模型 ms_out = MeanShift(bandwidth=3).fit(X_with_outliers) # 可视化时用不同符号标记离群点 labels_out = ms_out.labels_ plt.scatter(X_with_outliers[:,0], X_with_outliers[:,1], c=labels_out, s=15, edgecolor='k', alpha=0.7) plt.scatter(ms_out.cluster_centers_[:,0], ms_out.cluster_centers_[:,1], c='red', s=200, marker='X') plt.title('存在离群点时的聚类效果') plt.show()

!显示三个主簇外存在孤立点

这时候会发现那些孤立的点要么自成一组,要么被忽略——具体取决于带宽设置。这种特性在欺诈检测场景中反而成了优点,可以把异常交易自动筛出来。

最后给新人提个醒:虽然sklearn的实现很方便,但处理大规模数据时记得开启cluster_all=False参数,否则内存可能爆炸。工业级应用通常会结合空间索引进行优化,不过那是进阶玩法了。下次遇到形状不规则的客户分群需求,不妨先试试这个"随大流"的方法,说不定比K-means更管用。

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/6/15 9:31:31

使用Miniconda-Python3.9快速部署HuggingFace大模型

使用Miniconda-Python3.9快速部署HuggingFace大模型 在AI研发一线,你是否经历过这样的场景:刚从同事那里拿到一个基于HuggingFace的文本生成项目,兴冲冲地pip install -r requirements.txt,结果却因为PyTorch版本不兼容、CUDA驱动…

作者头像 李华
网站建设 2026/6/15 10:26:59

Pyenv local设置项目级Python版本

项目级 Python 版本管理与轻量 AI 开发环境构建 在现代软件开发中,尤其是人工智能、数据科学和 Web 工程领域,Python 的广泛应用带来了极大的灵活性,也引出了一个棘手的问题:如何在同一个系统上安全、高效地运行多个依赖不同 Pyth…

作者头像 李华
网站建设 2026/6/15 9:31:28

好写作AI|当AI邂逅古典文学:你的“赛博同窗”能解李杜诗篇吗?

当文科生面对《红楼梦》三千条脂批陷入沉思时,理科生已用AI写出了90分的《论黛玉葬花与当代生态女性主义》——这届文学论文,正在被智能工具悄悄改写。“老师,AI能读懂‘无边落木萧萧下’的悲凉吗?” “同学,你需要的不…

作者头像 李华
网站建设 2026/6/15 10:23:26

科研必备:用Miniconda构建可重复实验的AI开发环境

科研必备:用Miniconda构建可重复实验的AI开发环境 在人工智能科研一线,你是否经历过这样的场景:论文复现失败,模型跑不通,调试数日才发现问题出在某台机器上 numpy 是 1.23 版本,而另一台是 1.26&#xff1…

作者头像 李华
网站建设 2026/6/15 10:23:57

CondaError处理大全:从init到activate全过程排错

CondaError处理大全:从init到activate全过程排错 在现代 Python 开发中,尤其是在人工智能、数据科学和机器学习项目里,环境管理早已不是“可选项”,而是决定开发效率与实验复现性的核心环节。你有没有遇到过这样的场景&#xff1…

作者头像 李华