Focal_loss 的实现
TensorFlow 1.13版本
def focal_loss(logits,onehot_labels, gamma=2.0, alpha=4.0):
"""
focal loss for multi-classification
FL(p_t)=-alpha(1-p_t)^{gamma}ln(p_t)
Notice: logits is probability after softmax
gradient is d(Fl)/d(p_t) not d(Fl)
原创
2020-11-30 20:05:38 ·
666 阅读 ·
2 评论