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

2024-05-13 17:37:24 发布

您现在位置:Python中文网/ 问答频道 /正文

我是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_batch()所需的位置参数padded_shapes
我已经学会了如何使用函数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的提示吗

谢谢大家!


Tags: orgtestnonedatasize网站tensorflowbatch
1条回答
网友
1楼 · 发布于 2024-05-13 17:37:24

请参考下面的代码

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)

相关问题 更多 >