角膜上的分类聚焦损失

时间:2019-05-24 13:45:26

标签: tensorflow keras

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

2 个答案:

答案 0 :(得分:0)

好的。我通过搜索Keras focal loss找到了this。这是第一个结果,并且花费的时间更少。

This是Google上的第二个结果。也尝试过,它也很好用;参加了我的分类问题之一,最高得分达到0.9726。

Google是您的朋友。

答案 1 :(得分:0)

  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)