我有一个经典的y_train，它由一维维度的0（负）和1（正）组成。我想训练一个张量流模型，但是我必须用我想要的类数来初始化y占位符。所以在这种文本分类的情况下，我希望模型检查负值或正值，所以2类？但是我如何将y_train转换为适合我正在寻找的输出。谢谢你的时间！
"ValueError: Cannot feed value of shape (25000, 1) for Tensor u'Placeholder_5:0', which has shape (Dimension(None), Dimension(2))"
最佳答案
看来您的y_train本身包含标签值，而模型所需的y_train则需要标签概率。在您的情况下，由于只有两个标签，因此可以将其转换为标签概率，如下所示：
y_train = tf.concat(1, [1 - y_train, y_train])
如果您有更多标签，请查看
sparse_to_dense将其转换为概率。