news 2026/6/10 17:16:41

均值漂移聚类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/10 12:50:11

Miniconda中conda create创建PyTorch专用环境命令

Miniconda中conda create创建PyTorch专用环境命令 在深度学习项目开发中,一个常见的场景是:你刚接手一个基于 PyTorch 的代码库,兴冲冲地运行 python train.py,结果却抛出一堆 ModuleNotFoundError 或版本不兼容的错误。更糟的是&…

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

Miniconda中使用pip与conda混合安装PyTorch注意事项

Miniconda中使用pip与conda混合安装PyTorch注意事项 在深度学习项目开发中,环境配置往往是第一步,却也常常成为最令人头疼的环节。你是否遇到过这样的情况:代码写好了,依赖装上了,结果 torch.cuda.is_available() 返回…

作者头像 李华
网站建设 2026/6/10 2:10:21

Docker compose编排Miniconda服务与数据库联动

Docker Compose 编排 Miniconda 服务与数据库联动 在数据科学和 AI 工程实践中,最让人头疼的往往不是模型本身,而是“为什么代码在我机器上跑得好好的,换台环境就报错?”——依赖版本冲突、Python 环境混乱、数据库连接失败……这…

作者头像 李华
网站建设 2026/6/9 22:21:29

Jupyter Notebook魔法命令大全:%time %load

Jupyter Notebook魔法命令实战:深入解析%time与%load的高效用法 在数据科学和人工智能项目中,我们常常面临两个核心挑战:如何快速定位代码性能瓶颈?以及怎样高效复用已有逻辑避免重复造轮子?虽然Python生态提供了丰富的…

作者头像 李华
网站建设 2026/6/9 17:18:57

Linux iotop监控Miniconda磁盘IO性能

Linux iotop监控Miniconda磁盘IO性能 在现代AI与数据科学开发中,Python环境的稳定性与效率直接影响项目进度。一个看似简单的conda install命令背后,可能隐藏着数十个包的下载、解压、链接和缓存操作——这些动作都会转化为实实在在的磁盘I/O压力。当你的…

作者头像 李华
网站建设 2026/6/8 7:37:19

PyTorch安装教程GPU版本:基于Miniconda-Python3.9镜像快速部署

PyTorch GPU 环境部署实战:基于 Miniconda-Python3.9 镜像的高效搭建方案 在当前深度学习项目开发中,最让人头疼的往往不是模型设计本身,而是环境配置——明明本地跑得好好的代码,换一台机器就报错“CUDA not available”&#x…

作者头像 李华