python - 多热标签编码

Question

我是 Tensorflow 新手。我有一个图像数据集，其中一张图像有多个标签。据我了解，我需要使用 tf.losses.sigmoid_cross_entropy() 。我尝试将 tf.one_hot 应用于标签，但是当我尝试将它们传递到损失函数时，出现错误，形状不兼容。我该如何解决这个问题？

Answer 1

您对tf.losses.sigmoid_cross_entropy的看法是正确的。您所需要做的就是用 tf.reduce_max 包装 tf.one_hot 来降低维度。

tf.reduce_max(tf.one_hot(labels, num_classes, dtype=tf.int32), axis=0)

这应该返回形状为 (num_classes,) 的张量，这正是损失函数所需的。