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)