使用Tensorflow进行多标签分类

Question

我有5个标签的多标签分类问题。如果我的输出图层具有形状[n_samples，n_labels = 5]并且丢失函数为tf.nn.sigmoid_cross_entropy_with_logits，这是正确的吗？或者我应该将输出层设置为[n_samples，n_labels = 5,2]的形状，其中2是要编码，如果每个标签都分配给样本？

Answer 1

我认为输出图层的形状应为[n_samples，n_labels]，当您使用tf.nn.softmax_cross_entropy_with_logits时，请确保标签数组的格式为one-hot，另一个推荐的接口为{{3} }，哪些标签应该是n_labels的条目;