机器学习(二十二):精度和召回率

一、倾斜数据集

倾斜数据集:一个数据集中的正面和负面例子的比例非常不平衡,比如数据集中,结果为1的占比20%,结果为0的占比80%

例子:如果数据集的结果中只有0.5%是1,其余结果是0。有一个模型的预测准确度是99.5%,但是预测了所有数据的结果都是0,这个模型的准确度很高,但是预测不出结果为1,这不能代表这个模型是好模型。因此需要引入其他的误差度量方式来评估模型好坏。

二、精度和召回率

精度:预测为1的数据中,实际真正为1的占比。

召回率:实际为1的数据中,预测真正为1的占比。

例子:下图是实际结果为1和0时,预测结果为1和0的数量统计。

  • 实际为1,预测真1的数据有15个;
  • 实际为1,预测假0的数据有10个;
  • 实际为0,预测假1的数据有5个;
  • 实际为0,预测真0的数据有70个

精度计算:

精度=真正1的数量/预测为1的数量=真1的数量/(真1的数量+假1的数量)=15/(15+5)

召回率计算:

召回率=真正1的数量/实际1的数量=真1的数量/(真1的数量+假0的数量)=15/(15+10)

三、精度和召回的权衡

以逻辑回归为例:模型预测出结果为1的概率是

可以设定阈值为0.5,当≥0.5时,结果为1,当<0.5时,结果为0

也可以设定阈值为0.7,当≥0.7时,结果为1,当<0.7时,结果为0

也可以设定阈值为0.3,当≥0.3时,结果为1,当<0.3时,结果为0

当提高阈值,能提高精度,但是会降低召回率

当降低阈值,能提高召回率,但是会降低精度

如何权衡精度和召回率?

可以使用F1 score结合精度和召回率,F1 score也称为谐波平均值,是一种取平均值的方法,计算结果更偏向于较小的值。

计算公式(P为精度,R为召回):

学习来源:吴恩达机器学习,14.1-14.2节

相关推荐
张较瘦_4 分钟前
[论文阅读] AI | 用机器学习给深度学习库“体检”:大幅提升测试效率的新思路
论文阅读·人工智能·机器学习
杜子不疼.7 分钟前
CANN图引擎GE的编译优化与高效执行机制深度解析
人工智能·深度学习
池央7 分钟前
CANN 诊断工具链深度解析:oam-tools 的自动化故障信息收集、软硬件状态快照与 AI Core 错误溯源机制
运维·人工智能·自动化
深圳行云创新8 分钟前
采用 TitanIDE 3.0 开展团队级 AI-Coding 优势分析
人工智能
算法狗28 分钟前
大模型面试题:大模型的训练和推理中显存和计算量的情况
人工智能·深度学习·机器学习·语言模型
AI职业加油站12 分钟前
职业提升之路:我的大数据分析师学习与备考分享
大数据·人工智能·经验分享·学习·职场和发展·数据分析
风指引着方向14 分钟前
昇腾算子性能调优:ops-nn 中的内存布局与向量化技巧
java·大数据·人工智能
班德先生25 分钟前
以全案策划设计思维破局,让电器科技品牌力落地生根
大数据·人工智能·科技
ujainu26 分钟前
CANN仓库中的AIGC确定性推理工程:昇腾AI软件栈如何在混沌中构建“可预测的智能”
人工智能·aigc
咕泡科技27 分钟前
架构演进:从确定性工作流 (Workflow) 到自主智能体 (LLM Agent)
人工智能·架构