pytorch中的loss函數(1):MultiLabelSoftMarginLoss

MultiLabelSoftMarginLoss針對multi-label one-versus-all(多分類,且每個樣本只能屬於一個類)的情形。 loss的計算公式如下: 其中,x是模型預測的標籤,x的shape是(N,C),N表示batch size,C是分類數;y是真實標籤,shape也是(N,C),。 的值域是(0,); 的值域是(1,); 的值域是(0,1); 的值域是(-,0),函數
相關文章
相關標籤/搜索