TensorFlow 2.0 RC is available Learn more

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

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

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

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

  • fr-en (v0.0.3) (Size: 6.20 GiB): WMT 2014 fr-en translation task dataset.

  • hi-en (v0.0.3) (Size: 44.65 MiB): WMT 2014 hi-en translation task dataset.

  • ru-en (v0.0.3) (Size: 998.38 MiB): WMT 2014 ru-en translation task dataset.

wmt14_translate/cs-en

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

wmt14_translate/de-en

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

wmt14_translate/fr-en

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

wmt14_translate/hi-en

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

wmt14_translate/ru-en

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

Statistics

Split Examples
ALL 2,492,968
TRAIN 2,486,965
TEST 3,003
VALIDATION 3,000

Urls

Supervised keys (for as_supervised=True)

(u'ru', u'en')

Citation

@InProceedings{bojar-EtAl:2014:W14-33,
  author    = {Bojar, Ondrej  and  Buck, Christian  and  Federmann, Christian  and  Haddow, Barry  and  Koehn, Philipp  and  Leveling, Johannes  and  Monz, Christof  and  Pecina, Pavel  and  Post, Matt  and  Saint-Amand, Herve  and  Soricut, Radu  and  Specia, Lucia  and  Tamchyna, Ale
{s}},
  title     = {Findings of the 2014 Workshop on Statistical Machine Translation},
  booktitle = {Proceedings of the Ninth Workshop on Statistical Machine Translation},
  month     = {June},
  year      = {2014},
  address   = {Baltimore, Maryland, USA},
  publisher = {Association for Computational Linguistics},
  pages     = {12--58},
  url       = {http://www.aclweb.org/anthology/W/W14/W14-3302}
}