我有一个经典的 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
将它们转换为概率。
关于machine-learning - reshape y_train 以在 Tensorflow 中进行二进制文本分类,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/35198528/