检查 python 数据加载器的内容

Question

这可能是一个简单的问题，但是如何查看这个标准数据加载器的内容：

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] 或类似的东西......

Answer 1

数据集是可迭代的，您可以使用 获取第一个元素next(iter(test_data))。