Python PyTorch to_map_style_dataset用法及代码示例

本文简要介绍python语言中 torchtext.data.functional.to_map_style_dataset 的用法。

用法:

torchtext.data.functional.to_map_style_dataset(iter_data)

参数：

iter_data-一个迭代器类型对象。示例包括可迭代数据集、字符串列表、文本 io、生成器等。

将iterable-style 数据集转换为map-style 数据集。

例子

>>> from torchtext.datasets import IMDB
>>> from torchtext.data import to_map_style_dataset
>>> train_iter = IMDB(split='train')
>>> train_dataset = to_map_style_dataset(train_iter)
>>> file_name = '.data/EnWik9/enwik9'
>>> data_iter = to_map_style_dataset(open(file_name,'r'))