我正在尝试使用自己的数据遵循基本的word embedding tutorial。但是,当我尝试训练模型时,标题中遇到ValueError。我没有使用任何逻辑数组,所以找不到错误。 张量流模型 我正在传递模型的数据 任何帮助将不胜感激model = keras.Sequential([
layers.Embedding(vocab_size, embedding_dim),
layers.GlobalAveragePooling1D(),
layers.Dense(16, activation='relu'),
layers.Dense(1, activation='sigmoid')
])
model.summary()
model.compile(optimizer='adam',
loss='binary_crossentropy',
metrics=['accuracy'])
history = model.fit(
train_data,
epochs=10,
validation_data=test_data, validation_steps=20)
tf.Tensor(
[[ 83 97 106 ... 0 0 0]
[ 74 252 114 ... 0 0 0]
[ 66 108 97 ... 0 0 0]
...
[ 83 97 114 ... 0 0 0]
[ 67 97 109 ... 0 0 0]
[ 87 105 108 ... 0 0 0]], shape=(3000, 1999), dtype=int32)
0 个答案:
没有答案