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

TensorFlow模型怎么计算准确率_Python编写自定义Metrics指标

雨丽吖_3967

雨丽吖_3967

发布时间:2026-05-11 09:40:00

|

126人浏览过

|

来源于php中文网

原创

tf.keras.metrics.Accuracy训练时为0,因其默认逐元素比较logits与label且不自动argmax;应改用SparseCategoricalAccuracy(int标签)或CategoricalAccuracy(one-hot标签)。

tensorflow模型怎么计算准确率_python编写自定义metrics指标

tf.keras.metrics.Accuracy 为什么在训练时显示为 0?

常见现象是:模型编译时用了 tf.keras.metrics.Accuracy(),但训练中 accuracy 始终为 0,而 loss 正常下降。这是因为 Accuracy 默认对 logits 和 label 做逐元素比较,没经过 argmax —— 它期望输入已经是整数类别(如 [1, 0, 2]),但大多数分类模型输出的是 logits 或概率分布(如 [[2.1, -1.3, 0.8], ...])。

  • 解决方法:改用 tf.keras.metrics.SparseCategoricalAccuracy()(label 是 int 类型,logits 直接输入)或 tf.keras.metrics.CategoricalAccuracy()(label 是 one-hot,需先对 logits tf.nn.softmax)
  • 验证方式:打印 y_pred.shape 和 y_true.shape —— 若 y_true 是 shape=(N,) 的整数,就用 SparseCategoricalAccuracy;若是 shape=(N, C) 的 one-hot,就用 CategoricalAccuracy
  • 注意:Accuracy 类本身不自动做 argmax,它只做 y_pred == y_true,所以传 logits 过去几乎必然全错

如何写一个带阈值的二分类准确率 metric?

标准 BinaryAccuracy 允许设 threshold 参数,但有时你需要更细粒度控制,比如“预测概率 > 0.7 才算正类”,或想同时统计 TP/TN/FP/FN。这时自定义 metric 更灵活。

  • 继承 tf.keras.metrics.Metric,重写 __init__、update_state、result
  • update_state 中用 tf.cast(tf.greater(y_pred, 0.7), tf.int32) 得预测标签,再用 tf.equal 对比 y_true,最后用 self.total 和 self.count 累加(推荐用 tf.Variable + assign_add)
  • 别忘了在 reset_state 里重置变量,否则多轮 eval 会累积
  • 示例关键片段:
    class ThresholdedBinaryAccuracy(tf.keras.metrics.Metric):
      def __init__(self, threshold=0.7, name='thresholded_acc', **kwargs):
        super().__init__(name=name, **kwargs)
        self.threshold = threshold
        self.total = self.add_weight(name='total', initializer='zeros')
        self.count = self.add_weight(name='count', initializer='zeros')
    
      def update_state(self, y_true, y_pred, sample_weight=None):
        y_pred = tf.cast(tf.greater(y_pred, self.threshold), tf.int32)
        y_true = tf.cast(y_true, tf.int32)
        values = tf.cast(tf.equal(y_pred, y_true), self.dtype)
        self.total.assign_add(tf.reduce_sum(values))
        self.count.assign_add(tf.cast(tf.size(values), self.dtype))
    
      def result(self):
        return self.total / self.count

自定义 metric 在 model.fit() 里不更新?检查这几个点

写完自定义 metric,加到 model.compile(metrics=[...]) 后发现训练日志里压根不显示,或者值恒为 0/NaN,大概率是这几处出问题:

macOS UI 自动化技能包 —— 用 Python 控制你的 Mac
macOS UI 自动化技能包 —— 用 Python 控制你的 Mac

让 macOS 自动化像说话一样简单 —— 用代码控制你的 Mac,无需学习 AppleScript 系统要求 macOS 14+ (Sonoma) Python 3.9+ 需要开启:辅助功能 + 屏幕录制权限

下载
  • update_state 没正确处理 batch 维度:确保 y_true 和 y_pred 的 batch 维一致,避免因 broadcasting 导致 shape 错误(例如 y_true 是 (32,),y_pred 是 (32, 1) —— 要先 tf.squeeze)
  • 忘记调用 super().__init__() 或漏掉 add_weight,导致内部状态变量未注册
  • metric 返回值类型不匹配:result() 必须返回标量 tf.Tensor,且 dtype 通常为 tf.float32;若返回 Python float 或 numpy array,Keras 会静默失败
  • 使用了不可微或非图模式操作(如 print、np.array),尤其在 TF 2.x eager 关闭时会报错,建议全程用 tf.* 操作

多输出模型怎么给每个 head 配独立准确率?

当模型有多个输出(如主任务 + 辅助任务),直接传 list 给 metrics 会报错或错配。Keras 要求 metrics 的结构与 model 输出严格对齐。

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

  • 如果 model 输出是 dict(如 {'main': ..., 'aux': ...}),metrics 也必须是 dict:metrics={'main': tf.keras.metrics.SparseCategoricalAccuracy(), 'aux': tf.keras.metrics.SparseCategoricalAccuracy()}
  • 如果 model 输出是 list(如 [main_out, aux_out]),metrics 就得是 list,且长度一致:metrics=[tf.keras.metrics.SparseCategoricalAccuracy(), tf.keras.metrics.SparseCategoricalAccuracy()]
  • label 数据也要对应:list 输出时,y_train 必须是 [y_main, y_aux];dict 输出时,y_train 是 {'main': ..., 'aux': ...}
  • 注意:自定义 metric 若用于某一个 head,它的 update_state 只会收到对应位置的 y_true 和 y_pred,无需额外切片
TensorFlow 自定义 metric 最容易卡在张量形状和状态变量生命周期上,尤其是跨 batch 累积时忘了 reset 或维度没对齐。写完先用小 batch 手动调用 update_state + result 测一遍,比等整个 fit() 跑完再调试快得多。

热门AI工具

更多
音述AI
音述AI Hot

一款AI音频处理工具,主要用于音述AI是一个以“用声音述说故事”为核心的 AI 音乐创作与声音分享社区,适合需要提升相关任务效率的用户。

DeepSeek

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

讯飞绘文

讯飞绘文是一款由科大讯飞推出的一站式 AIGC 内容运营平台。

超级简历WonderCV

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

超级麦吉

一款通过数字员工协作完成调研、数据分析与文档制作的AI办公平台。

VibeKnow
VibeKnow Hot

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

豆包大模型

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

WorkBuddy

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

火山引擎

火山引擎是一款面向企业的云计算与AI服务平台。

相关专题

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

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

1851

2023.07.20

python能做什么
python能做什么

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

5004

2023.07.25

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

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

1789

2023.07.31

python教程
python教程

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

29997

2023.08.03

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

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

3567

2023.08.04

python eval
python eval

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

3587

2023.08.04

scratch和python区别
scratch和python区别

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

1223

2023.08.11

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

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

616

2023.08.10

Uvicorn 报错与问题解决教程
Uvicorn 报错与问题解决教程

本专题汇总了Uvicorn服务器常见问题与报错解决的教程合集,涵盖启动失败、端口占用、模块导入错误、事件循环冲突及热重载失效等典型故障的排查思路。同时整理了多进程与Worker配置、SSL/HTTPS证书报错、跨域与代理配置等实战解决方案,助你快速定位并解决Uvicorn运行中的各类问题。

0

2026.11.02

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
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