讲师中心 微信公众号
AI工具推荐 视频效率加速

如何使用Python中的Scikit-learn计算聚类算法的调整兰德系数?

浅芳酱_3705

浅芳酱_3705

发布时间:2026-06-15 12:18:04

|

1060人浏览过

|

来源于php中文网

原创

adjusted_rand_score用于评估聚类结果与真实标签的一致性,取值范围[-1,1],1表示完全一致,0表示等价于随机划分;要求两组整数标签长度一致、无NaN、为一维数组,对簇名顺序不敏感。

如何使用python中的scikit-learn计算聚类算法的调整兰德系数?

调整兰德系数(Adjusted Rand Index, ARI)在 scikit-learn 中直接可用,但必须确保真实标签和聚类结果标签都为整数数组且长度一致;否则会报 ValueError: Input contains NaN, infinity or a value too large for dtype('float64') 或 labels_true and labels_pred must have same size。

用 adjusted_rand_score 函数计算 ARI

scikit-learn 提供了 sklearn.metrics.adjusted_rand_score,它接受两个一维数组:真实类别标签 labels_true 和模型预测的聚类标签 labels_pred。该函数不关心标签具体取值(如 0/1/2 或 5/10/15),只看划分一致性。

常见错误现象包括:

  • 传入 list 嵌套结构(如 [[0], [1], [0]])而非一维数组 —— 需用 np.ravel() 或 .flatten() 展平
  • 聚类结果含 NaN 或 None(例如某些 DBSCAN 把噪声点标为 -1 是合法的,但若误填 np.nan 就会报错)
  • 使用 fit_predict 返回的是 ndarray,但真实标签是 pandas.Series,索引未对齐导致长度看似一致实则错位

示例:

立即学习“Python免费学习笔记(深入)”;

from sklearn.metrics import adjusted_rand_score
import numpy as np
<p>labels_true = [0, 0, 1, 1, 2, 2]
labels_pred = [0, 0, 1, 1, 1, 1]  # 比如 KMeans 把后两类全分进同一簇</p><p>score = adjusted_rand_score(labels_true, labels_pred)
print(score)  # 输出类似 0.24...

ARI 对标签编码方式不敏感,但对空簇和噪声点很敏感

adjusted_rand_score 基于配对计数(pair counting),所以只要两个标签序列的“哪些样本被分到同一组”逻辑一致,分数就相同。这意味着:

Forge — Python Developer
Forge — Python Developer

Python开发助手 — 自动化脚本、API封装、命令行工具、数据管道、网页爬虫。经过测试的代码,包含错误处理和文档。当你需要...

下载
  • [0, 0, 1, 1] 和 [5, 5, 99, 99] 与真实标签比,ARI 完全一样
  • DBSCAN 返回的 -1 表示噪声点,这是合法标签值,会被正常计入计算 —— 无需提前过滤或替换
  • 如果某簇完全没样本(比如 KMeans 设 n_clusters=4 但实际只产生 3 个非空簇),不影响 ARI,因为函数只看已有标签的配对关系

但注意:若 labels_pred 全是同一个值(如全为 0),而 labels_true 有多个类,则 ARI 会是负数(表示比随机划分还差),这是正常行为,不是 bug。

和 homogeneity_score、completeness_score 的关键区别

ARI 是对称指标,不偏向“真实标签”或“预测标签”哪一方;而 homogeneity_score 要求每个簇纯度高,completeness_score 要求同类样本尽量不分散 —— 它们都不做随机性校正,数值范围是 [0,1],而 ARI 是 [-1,1],0 表示等价于随机划分。

典型使用场景:

  • 比较不同聚类算法在同一数据集上的效果(如 KMeans vs AgglomerativeClustering)
  • 调参时评估 n_clusters 取值 —— 注意 ARI 在真实类数附近常出现峰值,但未必严格对应
  • 当真实标签存在但不完全可靠(如人工标注有少量错误),ARI 比纯精度更鲁棒,因它关注样本对关系而非逐点匹配

性能影响很小:adjusted_rand_score 时间复杂度是 O(n),n 是样本数;内部用稀疏矩阵计数,10⁵ 级样本也能秒级完成。

最容易被忽略的一点:ARI 无法处理多标签分类或层次化聚类输出;如果你的 labels_pred 是树状结构或每个样本属于多个簇,得先转换成单标签形式(如取最大概率簇),否则会直接报错或结果无意义。

热门AI工具

更多
WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

PixPix
PixPix Hot

PixPix是一款面向电商视觉生产的AI商品图生成工具。

LibLibAI
LibLibAI Hot

一款AI视频创作工具,主要用于国内领先的AI创意平台,以海量模型、低门槛操作与“创作-分享-商业化”生态,让小白与专业创作者都能高效实现图文乃至视频创意表达,适合需要提升相关任务效率的用户。

VibeKnow
VibeKnow Hot

一款AI视频创作工具,主要用于全球首个AI知识视频创作平台,文档、文章、网页,一键生成视频,适合需要提升相关任务效率的用户。

Seko
Seko Hot

一款AI视频创作工具,主要用于商汤科技推出的创编一体的AI短视频创作Agent,适合需要提升相关任务效率的用户。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

超级简历WonderCV

一款AI办公效率工具,主要用于免费求职简历模版下载制作,应届生职场人必备简历制作神器,适合需要提升相关任务效率的用户。

讯飞智作

讯飞智作是一款AI视频创作工具,AI文本配音工具,数字人课程、营销视频制作。

相关专题

更多
python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

1731

2023.07.20

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

4444

2023.07.25

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

1709

2023.07.31

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

26057

2023.08.03

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

3167

2023.08.04

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

3187

2023.08.04

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

1183

2023.08.11

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

596

2023.08.10

Linux 新手先学哪些命令
Linux 新手先学哪些命令

Linux新手命令专题汇集终端基础、文件目录操作、文本查看、软件安装和权限管理等入门文章,帮助按实际任务练习常用命令并建立命令行操作习惯。

0

2026.10.15

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Uvicorn官方安装与入门手册
Uvicorn官方安装与入门手册

共0课时 | 0人学习

Python uv官方项目管理指南
Python uv官方项目管理指南

共0课时 | 0人学习

Python uv官方安装与入门文档
Python uv官方安装与入门文档

共0课时 | 0人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn