Проверка содержимого python загрузчик данных - PullRequest
0 голосов
/ 19 июня 2020

Вероятно, это простой вопрос, но как посмотреть, как выглядит содержимое этого стандартного загрузчика данных:

from torchtext import datasets
import random

train_data, test_data = datasets.IMDB.splits(TEXT, LABEL)
train_data, valid_data = train_data.split(random_state = random.seed(SEED))

Я не могу использовать .head () и:

print(test_data)

просто дает мне:

<torchtext.datasets.imdb.IMDB object at 0x7f0b42e8c240>

Вероятно, мне просто не хватает .values ​​или [0] или чего-то подобного ...

1 Ответ

2 голосов
/ 19 июня 2020

Наборы данных являются повторяемыми, вы можете получить первый элемент с помощью next(iter(test_data)).

...