如何从tf.dataset打印样本记录？

Question

我有一个名为imdb的tensorflow数据集。如何打印此数据集的前5条记录(包括标头)？我正在寻找类似于熊猫dataframe.head()的东西。一些数据集也可以有图像。

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

代码print(type(imdb))返回<class 'dict'>。我知道tfds.show_examples只支持图像。谢谢。

Answer 1

假设您已经启用了急切的执行，下面将显示前五个示例：

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