TensorFlow 2.0 RC is available Learn more

wmt17_translate

Translate dataset based on the data from statmt.org.

Versions exists for the different years using a combination of multiple data sources. The base wmt_translate allows you to create your own config to choose your own data/language pair by creating a custom tfds.translate.wmt.WmtConfig.

config = tfds.translate.wmt.WmtConfig(
    version="0.0.1",
    language_pair=("fr", "de"),
    subsets={
        tfds.Split.TRAIN: ["commoncrawl_frde"],
        tfds.Split.VALIDATION: ["euelections_dev2019"],
    },
)
builder = tfds.builder("wmt_translate", config=config)

wmt17_translate is configured with tfds.translate.wmt17.WmtConfig and has the following configurations predefined (defaults to the first one):

  • cs-en (v0.0.3) (Size: 1.66 GiB): WMT 2017 cs-en translation task dataset.

  • de-en (v0.0.3) (Size: 1.81 GiB): WMT 2017 de-en translation task dataset.

  • fi-en (v0.0.3) (Size: 414.10 MiB): WMT 2017 fi-en translation task dataset.

  • lv-en (v0.0.3) (Size: 161.69 MiB): WMT 2017 lv-en translation task dataset.

  • ru-en (v0.0.3) (Size: 3.34 GiB): WMT 2017 ru-en translation task dataset.

  • tr-en (v0.0.3) (Size: 59.32 MiB): WMT 2017 tr-en translation task dataset.

  • zh-en (v0.0.3) (Size: 2.16 GiB): WMT 2017 zh-en translation task dataset.

wmt17_translate/cs-en

Translation({
    'cs': Text(shape=(), dtype=tf.string),
    'en': Text(shape=(), dtype=tf.string),
})

wmt17_translate/de-en

Translation({
    'de': Text(shape=(), dtype=tf.string),
    'en': Text(shape=(), dtype=tf.string),
})

wmt17_translate/fi-en

Translation({
    'en': Text(shape=(), dtype=tf.string),
    'fi': Text(shape=(), dtype=tf.string),
})

wmt17_translate/lv-en

Translation({
    'en': Text(shape=(), dtype=tf.string),
    'lv': Text(shape=(), dtype=tf.string),
})

wmt17_translate/ru-en

Translation({
    'en': Text(shape=(), dtype=tf.string),
    'ru': Text(shape=(), dtype=tf.string),
})

wmt17_translate/tr-en

Translation({
    'en': Text(shape=(), dtype=tf.string),
    'tr': Text(shape=(), dtype=tf.string),
})

wmt17_translate/zh-en

Translation({
    'en': Text(shape=(), dtype=tf.string),
    'zh': Text(shape=(), dtype=tf.string),
})

Statistics

Split Examples
ALL 25,140,612
TRAIN 25,136,609
VALIDATION 2,002
TEST 2,001

Urls

Supervised keys (for as_supervised=True)

(u'zh', u'en')

Citation

@InProceedings{bojar-EtAl:2017:WMT1,
  author    = {Bojar, Ond
{r}ej  and  Chatterjee, Rajen  and  Federmann, Christian  and  Graham, Yvette  and  Haddow, Barry  and  Huang, Shujian  and  Huck, Matthias  and  Koehn, Philipp  and  Liu, Qun  and  Logacheva, Varvara  and  Monz, Christof  and  Negri, Matteo  and  Post, Matt  and  Rubino, Raphael  and  Specia, Lucia  and  Turchi, Marco},
  title     = {Findings of the 2017 Conference on Machine Translation (WMT17)},
  booktitle = {Proceedings of the Second Conference on Machine Translation, Volume 2: Shared Task Papers},
  month     = {September},
  year      = {2017},
  address   = {Copenhagen, Denmark},
  publisher = {Association for Computational Linguistics},
  pages     = {169--214},
  url       = {http://www.aclweb.org/anthology/W17-4717}
}