keras 上的分类焦点损失

Question

我想在 tf/keras 中使用分类焦点损失。 Binary Focal loss 对我有用，但不适用于我为分类 f.l 找到的代码。有人有这个吗？

Answer 1

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

• binary_focal_loss

• sparse_categorical_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)
  

Answer 2

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

This 是谷歌上的第二个结果。也试过了，效果也不错；将我的一个分类问题提高到 0.9726 的 roc 分数。

Google 是你的朋友。