我有 5 个标签的多标签分类问题。如果我的输出层的形状为 [n_samples, n_labels=5] 并且损失函数为 tf.nn.sigmoid_cross_entropy_with_logits 是否正确？或者我应该有 [n_samples, n_labels=5, 2] 形状的输出层，其中 2 用于编码是否每个标签都分配给样本？
问问题
3381 次
1 回答
-2
我认为输出层的形状应该是 [n_samples, n_labels]，并且在使用 tf.nn.softmax_cross_entropy_with_logits 时，请确保标签数组是 one-hot 的格式，另一个推荐的接口是tf.nn.sparse_softmax_cross_entropy_with_logits， labels 应该是 n_labels 的入口；
于 2017-01-19T00:47:00.517 回答