如何轉換為多分類模型？

Question

我正在嘗試使用雲樣本github實現多類別分類，這是一個分類模型，我必須更改代碼。我發現了一些建議，可以將最終層和損失從softmax更改為Sigmoid。此外，我還必須更改標簽到一種熱編碼。有人可以幫助將標簽更改為一種熱編碼嗎？

Answer 1

請參考以下文章：

https://towardsdatascience.com/multi-label-image-classification-with-inception-net-cbb2ee538e30

該文章中使用的技術如下所示：

ground_truth = np.zeros(class_count, dtype=np.float32)
ground_truth[label_index] = 1.0

對於很多類（成千上萬），這可能無法很好地擴展。 要擴展到更多的類，您需要與sigmoid等效的tf.nn.sparse_softmax_cross_entropy_with_logits ，這似乎並不存在。