0

如何从 PyTorch Dataset(<torchtext.data.dataset.Dataset>) Object 获取训练数据作为文本(或文本列表)?

或者更具体地说,来自 torchtext.datasets.IMDB


例如来自torchtext.datasets (IMDB)

# make splits for data
train, test = datasets.IMDB.splits(TEXT, LABEL)

假设我在文档中完成了所有步骤,直到这里,我如何才能将 Dataset Object 中的第一个文本train作为元组以(“review”,分类)的形式获取:

print(first_review)
Output:
('This is a great movie',  1)
4

0 回答 0