题目给出了二分类场景下的 Focal Loss 定义:
你需要实现一个函数,用于计算二分类任务中的 Focal Loss。
给定模型预测概率 p∈(0,1)(通常为 Sigmoid 输出)以及真实标签 y∈{0,1},请你根据下述公式计算最终的 Focal Loss。
Focal Loss 定义为:
FL(p,y)=−α(1−p)γylog(p)−(1−α)pγ(1−y)log(1−p)其中:
若 y=1:
FL=−α(1−p)γlog(p)若 y=0:
FL=−(1−α)pγlog(1−p)输入参数
p: 模型预测概率。y: 真实标签(0 或 1)。alpha: Focal Loss 平衡系数。gamma: 调节困难样本权重的指数。返回值
loss: 最终的 Focal Loss。输入:
alpha = 0.25
gamma = 2
p = 0.7
y = 1
输出:
loss = 0.0080
Scan the QR code below with WeChat to sign in
First-time scan will create your account automatically
请使用微信扫描下方二维码完成注册