Reshape y_train 用于 Tensorflow 中的二进制文本分类

Reshape y_train for binary text classification in Tensorflow

我有一个经典的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 以将它们转换为概率。