Как распечатать образцы записей из tf.dataset?

Question

У меня есть набор данных тензорного потока под названием imdb. Как я могу напечатать 5 первых записей этого набора данных (включая заголовок)? Ищу что-то похожее на dataframe.head() в pandas. Некоторые наборы данных также могут содержать изображения.

   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)