[問題] 多類別問題

[jasonfghx (環宇 芯繫天下)]

如果在一張圖有很多類別項

例如

        類別1    類別2    類別3    類別4

圖A     1        1        0        0

圖B     0        0        0        1

圖C     0        1        1        0

圖D     0        1        0        0

變成像圖A具備類別1及2

然後用這樣概念去train 模型

然後預測時像圖C能至少預測出類別2及3機率最大這樣

請問像這類型問題該怎麼下關鍵字比較找的到東西

因為我找multi label是找到四個類別但都是分類到其中一個

謝謝答覆

--
※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 140.116.234.34
※ 文章網址: https://www.ptt.cc/bbs/Python/M.1536330534.A.635.html
→ Mchord: 直接當regression算L2 loss，activation 用sigmoid
→ jackwang01: 我記得搜尋multi label classification就很多資料了
→ jackwang01: 用sigmoid 針對每個class都去算binary cross entropy
→ jackwang01: 也行
→ jasonfghx: thx