首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >Tensorflow中的平衡精度分数

Tensorflow中的平衡精度分数
EN

Stack Overflow用户
提问于 2019-12-15 05:59:50
回答 4查看 3.5K关注 0票数 6

我正在为一个高度不平衡的分类问题实现一个CNN,我想在tensorflow中实现客户指标,以使用Select Best Model回调。具体地说,我想实现平衡的准确率分数,这是每个类的召回率的平均值(参见sklearn implementation here),有人知道如何做到这一点吗?

EN

Stack Overflow用户

发布于 2020-03-17 04:44:16

我还没有测试过这段代码,,但是看看tensorflow==2.1.0source code,这可能适用于二进制分类的情况:

代码语言:javascript
运行
复制
from tensorflow.keras.metrics import Recall
from tensorflow.python.ops import math_ops


class BalancedBinaryAccuracy(Recall):
    def result(self):
        result = (math_ops.div_no_nan(self.true_positives, self.true_positives + self.false_negatives) +
                  math_ops.div_no_nan(self.true_negatives, self.true_negatives + self.false_positives)) / 2
        return result[0] if len(self.thresholds) == 1 else result
票数 1
EN
查看全部 4 条回答
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/59339531

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档