Python PyTorch IWSLT2017用法及代码示例

本文简要介绍python语言中 torchtext.datasets.IWSLT2017 的用法。

用法:

torchtext.datasets.IWSLT2017(root='.data', split=('train', 'valid', 'test'), language_pair=('de', 'en'))

参数：

• root-保存数据集的目录。默认值：“.data”

• split-要返回的拆分或拆分。可以是字符串或字符串元组。默认值：(‘train’, ‘valid’, ‘test’)

• language_pair-包含 src 和 tgt 语言的元组或列表

IWSLT2017 数据集

可用的数据集包括：

语言对：

	‘en’	‘nl’	‘de’	‘it’	‘ro’
‘en’		x	x	x	x
‘nl’	x		x	x	x
‘de’	x	x		x	x
‘it’	x	x	x		x
‘ro’	x	x	x	x

有关更多详细信息，请参阅源网站：https://wit3.fbk.eu/2017-01

例子

>>> from torchtext.datasets import IWSLT2017
>>> train_iter, valid_iter, test_iter = IWSLT2017()
>>> src_sentence, tgt_sentence = next(train_iter)