小编fre*_*ata的帖子

keras中是否有基于精度或召回率而不是损失的优化程序?

我正在开发一个只有两个类别的分段神经网络,即0和1(0是背景,而1是我想在图像上找到的对象)。在每个图像上,大约1的80%和0的20%。如您所见,数据集是不平衡的,并且会导致结果错误。我的准确度是85%,损失很低,但这仅仅是因为我的模型善于寻找背景!

我想将优化器基于另一个指标,例如精度或召回率,在这种情况下更有用。

有人知道如何实现吗?

metrics machine-learning keras

7
推荐指数
3
解决办法
2076
查看次数

标签 统计

keras ×1

machine-learning ×1

metrics ×1