keras 角膜上的分类病灶丢失

gmol1639  于 2023-05-23  发布在  其他
关注(0)|答案(3)|浏览(239)

我想在tf/keras中使用分类焦点丢失。二进制焦点损失为我工作,但不是我找到的分类f.l.的代码。有人有这个吗?

8hhllhi2

8hhllhi21#

1.焦点损失0.0.6 https://pypi.org/project/focal-loss/
包括以下功能

  • 双焦损失
  • 稀疏分类焦点损失
pip install focal-loss
  1. https://github.com/umbertogriffo/focal-loss-keras
  • 二进制
model.compile(loss=[binary_focal_loss(alpha=.25, gamma=2)], metrics=["accuracy"], optimizer=adam)
  • 分类的
model.compile(loss=[categorical_focal_loss(alpha=[[.25, .25, .25]], gamma=2)], metrics=["accuracy"], optimizer=adam)
gupuwyp2

gupuwyp22#

当然。我是通过谷歌搜索Keras focal loss找到this的。这是第一个结果,并且花了更少的时间来实现。
This是Google上的第二个结果。也试过了,也工作得很好;把我的一个分类问题的ROC得分提高到0.9726。
Google是你的朋友。

gupuwyp2

gupuwyp23#

分类焦点丢失现在在tf.keras.losses.CategoricalFocalCrossentropy()下可用。您可以查看documentation以了解详细信息。

相关问题