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'))