1. 程式人生 > >處理不平衡類別的loss

處理不平衡類別的loss

在進行多分類問題的時候,常常會遇到資料不平衡的狀況,對於多分類問題,常見的loss有categorical crossentroy,可以考慮手動設定loss權重來增加模型的準確度,將類別較少的類權重增高。在試驗中,用的資料是情感分析label2的資料集,在不加權重f1值為0.53,權重為[1 2 3 1]後f1值為0.55, 權重為[1 2 6 1]後f1值為0.567。