如何转换为多分类模型？

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 ，这似乎并不存在。