我有一个经典的y_train,它由一维形状的0(负面)和1(正面)组成。我想训练一个tensorflow模型,但我必须用我想使用的类别数量来初始化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
来将它们转换为概率。