python - 如何在tensorflow.org的文本分类教程中定义函数padded_batch()的位置参数padded_shapes？

Question

我是 tensorflow 的新手，现在正在学习 tensorflow.org 的教程。这是我正在学习的教程的网站：https ://www.tensorflow.org/tutorials/keras/text_classification

但是当我运行下面的代码时它不起作用：

BBUFFER_SIZE = 1000

train_batches = (
train_data
.shuffle(BUFFER_SIZE)
.padded_batch(32))

test_batches = (
test_data
.padded_batch(32))

似乎问题在于没有给出所需的位置参数padded_shapes 。 我已经学会了如何使用该功能并尝试了几种方法，例如or ，但它仍然不起作用。相反，它会引发以下错误：padded_batch()
padded_batch()padded_shapes=[None]padded_shapes=([None], [None])

ValueError: The padded shape (None,) is not compatible with the corresponding input component shape ().

所以我现在不知道如何找出问题所在。有没有人学过这个教程，你是怎么解决的？有人可以给我一些关于如何定义的提示吗padded_shapes吗？

谢谢！

标签： pythontensorflowtensorflow-datasets

Answer 1

请参考以下代码

padded_shapes = ([None],()) 
train_batches = train_data.shuffle(1000).padded_batch(32,padded_shapes=padded_shapes)
test_batches = test_data.shuffle(1000).padded_batch(32,padded_shapes=padded_shapes)