machine-learning - reshape y_train 以在 Tensorflow 中进行二进制文本分类

Question

我有一个经典的 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))"

Answer 1

看来您的 y_train 包含标签值本身，而模型所需的 y_train 需要标签概率。在您的情况下，由于只有两个标签，您可以将其转换为标签概率，如下所示:

y_train = tf.concat(1, [1 - y_train, y_train])

如果您有更多标签，请查看sparse_to_dense将它们转换为概率。