小编Blu*_*row的帖子

我对 ML 比较陌生，以为我会从 keras 开始。在这里，我使用二元交叉熵将电影评论分类为正面或负面。因此，当我尝试使用 tensorflow estimator 包装我的 keras 模型时，出现错误：

Tensorflow 估计器 ValueError：logits 和标签必须具有相同的形状 ((?, 1) vs (?,))

我使用 sigmoid 激活作为我的最后一层，我猜我在这里遗漏了一些微不足道的东西。有什么帮助吗？

from tensorflow import keras
import tensorflow as tf
print("Tensorflow {} loaded".format(tf.__version__))
import numpy as np

keras.__version__
from keras.datasets import imdb

(train_data, train_labels), (test_data, test_labels) = imdb.load_data(num_words=10000)
def vectorize_sequences(sequences, dimension=10000):
    # Create an all-zero matrix of shape (len(sequences), dimension)
    results = np.zeros((len(sequences), dimension))
    for i, sequence in enumerate(sequences):
        results[i, sequence] = 1.  # set specific indices of results[i] to 1s …