当前位置:首页 > 技术 > 正文内容

分类模型评估指标:混淆矩阵、ROC与AUC解析

访客 技术 2026年7月11日 3

模型评估概述

模型评估用于选择最优模型、参数和特征组合。分类问题常用评估指标包括分类准确率、混淆矩阵及其衍生指标、ROC曲线和AUC值。

分类准确率的局限性

以皮马印第安人糖尿病数据集为例,使用逻辑回归模型:

import pandas as pd
from sklearn.model_selection import train_test_split
from sklearn.linear_model import LogisticRegression

diabetes_data = pd.read_csv('diabetes_dataset.csv')
features = ['glucose', 'bmi', 'age', 'pedigree']
X = diabetes_data[features]
y = diabetes_data.diabetes_status

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.25)

model = LogisticRegression(max_iter=200)
model.fit(X_train, y_train)
predictions = model.predict(X_test)

accuracy = model.score(X_test, y_test)
print(f"模型准确率: {accuracy:.3f}")

空准确率(多数类占比)计算:

majority_class = y_test.value_counts().index[0]
null_acc = (y_test == majority_class).mean()
print(f"空准确率: {null_acc:.3f}")

当准确率接近空准确率时,表明模型可能仅能识别多数类。

混淆矩阵分析

混淆矩阵提供更详细的分类表现:

from sklearn.metrics import confusion_matrix
cm = confusion_matrix(y_test, predictions)
print("混淆矩阵:\n", cm)

关键指标计算:

TP = cm[1, 1]  # 真阳性
TN = cm[0, 0]  # 真阴性
FP = cm[0, 1]  # 假阳性
FN = cm[1, 0]  # 假阴性

sensitivity = TP / (TP + FN)  # 召回率
specificity = TN / (TN + FP)  # 特异度
precision = TP / (TP + FP)    # 精确率
f1_score = 2 * (precision * sensitivity) / (precision + sensitivity)

不同场景的指标选择:

  • 欺诈检测:优先优化召回率(减少漏报)
  • 垃圾邮件过滤:优先优化精确率(减少误报)

阈值调整策略

通过调整分类阈值优化模型表现:

probabilities = model.predict_proba(X_test)[:, 1]
adjusted_predictions = (probabilities > 0.35).astype(int)

new_cm = confusion_matrix(y_test, adjusted_predictions)
new_recall = new_cm[1, 1] / (new_cm[1, 1] + new_cm[1, 0])
print(f"调整后的召回率: {new_recall:.3f}")

阈值降低会提高召回率但降低特异度,需根据业务需求平衡。

ROC与AUC评估

ROC曲线展示不同阈值下的性能:

from sklearn.metrics import roc_curve, roc_auc_score
import matplotlib.pyplot as plt

fpr, tpr, cutoffs = roc_curve(y_test, probabilities)
plt.plot(fpr, tpr)
plt.xlabel('假阳性率 (1 - 特异度)')
plt.ylabel('真阳性率 (召回率)')
plt.title('ROC曲线')
plt.show()

auc_value = roc_auc_score(y_test, probabilities)
print(f"AUC值: {auc_value:.3f}")

AUC值提供模型整体性能评估,适用于类别不平衡场景。值越接近1表示模型区分能力越强。

相关文章

Linux crontab 详解

1) crontab 是什么cron 是 Linux 的定时任务守护进程;crontab 是用来编辑/查看“按时间周期执行命令”的表(cron table)。常见两类:用户 crontab:每个用户一份(crontab -e 编辑)系统级 crontab / cron.d:可指定执行用户(/etc/crontab、/etc/cron.d/*)2) crontab 时间...

富文本里可以允许的 HTML 属性

一、所有标签默认允许的安全属性(极少)class        (可选)id           (通常建议禁用)title️ 注意:id 容易被滥用做锚点注入,很多系统直接禁用class 允许的话最好只允许固定前缀(如 editor-*)二、a 标签允许属性<a href="" t...

Mac 安装 Node.js 指南

方法一:通过官网安装包(最简单,适合初学者)如果你只是想快速安装并开始使用,这是最直接的方法。访问 Node.js 官网。页面会显示两个版本:LTS (Recommended For Most Users):长期支持版,最稳定。建议选这个。Current:最新特性版,包含最新功能但可能不够稳定。下载 .pkg 安装包并运行。按照安装向导点击“下一步”即可完成。方法二:使用 Homebrew 安装(...

Dom\HTML_NO_DEFAULT_NS 的副作用:自动加闭合标签

在使用Dom\HTMLDocument时,Dom\HTML_NO_DEFAULT_NS 将禁止在解析过程中设置元素的命名空间, 此设置是为了与DOMDocument向后兼容而存在的。当使用它时,已知的一个副作用就是:自动加闭合标签例如 </img> 为什么会这样?当你使用:Dom\HTML_NO_DEFAULT_NS文档会变成 无命名空间模式,此时内部更接近 XML...

Laravel 事件和监听器创建

在 Laravel 中,使用 Artisan 命令创建 Events(事件) 和 Listeners(监听器) 是非常高效的。你可以通过以下几种方式来实现:1. 手动创建单个 Event如果你只想创建一个事件类,可以使用 make:event 命令:Bashphp artisan make:event UserRegistered执行后,文件将生成在 app/Even...

自定义域名解析神器 dnsmasq

什么是 dnsmasq?dnsmasq 是一个轻量级、功能强大的网络服务工具,专为小型和中等规模网络设计。它是一个综合的网络基础设施解决方案[1]。dnsmasq 能做什么?功能说明应用场景DNS 转发与缓存将 DNS 查询转发到上游服务器(ISP、Google DNS 等),并在本地缓存结果加快 DNS 查询速度,减少外部 DNS 流量本地 DNS解析本地网络设备的主机名,无需编辑&n...

发表评论

访客

◎欢迎参与讨论,请在这里发表您的看法和观点。