lightgbm(xgboost)中的焦点损失和标签平滑,用于多类
此损失函数包含焦点损失[1]和标签平滑[2],现在仅支持lightgbm用于多类(类> 3,它将在以后支持xgboost和二进制类问题)
标签平滑
平滑值表示ε的值:
焦点损失
其中alpha,gamma是焦距损失的参数,为:
alpha用于不平衡样本,gamma用于难以学习的样本,在多类问题中,alpha似乎毫无用处。
用法:
1.导入损失函数库
import lightgbm as lgb
import lossfunction as lf
import numpy as np
2.初始化损失函数
focal_loss_lgb = lf.ComplexLoss(gamma = 0.5)
param_dist= {'objective':focal_loss_lgb.focal_loss}
param_dist['nu
2021-08-24 17:36:36
2KB
Python
1