如何打印来自 tf.dataset 的示例记录？

Question

我有一个名为 imdb 的张量流数据集。如何打印此数据集的前 5 条记录（包括 header）？我在 pandas 中寻找类似于 dataframe.head() 的内容。一些数据集也可以有图像。

   import tensorflow_datasets as tfds
   imdb, info = tfds.load("imdb_reviews", with_info=True, as_supervised=True)

代码print(type(imdb))returns<class 'dict'>。我了解 tfds.show_examples 仅支持图像。谢谢你。

Answer 1

假设您启用了即时执行，这将显示前五个示例：

for x in imdb['train'].take(5):
    print(x)