keras 上的分类焦点损失

Question

我想在 tf/keras 中使用分类焦点损失。Binary Focal loss 对我有用，但不是我为分类 fl 找到的代码 有人有吗？

Answer 1

1. focus-loss 0.0.6 https://pypi.org/project/focal-loss/
   包含以下功能

• 二进制焦点损失

• 稀疏分类焦点损失

  pip install focal-loss
  

2. 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)
  

Answer 2

当然。我通过谷歌搜索找到了这个Keras focal loss。这是第一个成果，实施的时间甚至更短。

这是谷歌的第二个结果。也尝试了一下，效果也很好；将我的一个分类问题的 roc 分数提高到 0.9726。

谷歌是你的朋友。