Question

我正在尝试使用cloud samples github进行多类分类。这是一个分类模型，我不得不更改代码。我发现了一些建议，可以将最终层和损失从softmax更改为Sigmoid。我必须将标签更改为一种热编码。有人可以帮助将标签更改为一种热编码吗？预先感谢

Answer 1

请参阅以下文章：

该文章中使用的技术如下：

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

对于很多类（成千上万），这可能无法很好地扩展。为了扩展到更多的类，您需要与Sigmoid等效的tf.nn.sparse_softmax_cross_entropy_with_logits，它似乎并不存在。