- Description:
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)
Additional Documentation: Explore on Papers With Code
Source code:
tfds.translate.Wmt16Translate
Versions:
1.0.0
(default): No release notes.
Manual download instructions: This dataset requires you to download the source data manually into
download_config.manual_dir
(defaults to~/tensorflow_datasets/downloads/manual/
):
Some of the wmt configs here, require a manual download. Please look into wmt.py to see the exact path (and file name) that has to be downloaded.Figure (tfds.show_examples): Not supported.
Citation:
@InProceedings{bojar-EtAl:2016:WMT1,
author = {Bojar, Ond
{r}ej and Chatterjee, Rajen and Federmann, Christian and Graham, Yvette and Haddow, Barry and Huck, Matthias and Jimeno Yepes, Antonio and Koehn, Philipp and Logacheva, Varvara and Monz, Christof and Negri, Matteo and Neveol, Aurelie and Neves, Mariana and Popel, Martin and Post, Matt and Rubino, Raphael and Scarton, Carolina and Specia, Lucia and Turchi, Marco and Verspoor, Karin and Zampieri, Marcos},
title = {Findings of the 2016 Conference on Machine Translation},
booktitle = {Proceedings of the First Conference on Machine Translation},
month = {August},
year = {2016},
address = {Berlin, Germany},
publisher = {Association for Computational Linguistics},
pages = {131--198},
url = {http://www.aclweb.org/anthology/W/W16/W16-2301}
}
wmt16_translate/cs-en (default config)
Config description: WMT 2016 cs-en translation task dataset.
Download size:
1.57 GiB
Dataset size:
7.56 GiB
Auto-cached (documentation): No
Splits:
Split | Examples |
---|---|
'test' |
2,999 |
'train' |
52,335,651 |
'validation' |
2,656 |
- Feature structure:
Translation({
'cs': Text(shape=(), dtype=string),
'en': Text(shape=(), dtype=string),
})
- Feature documentation:
Feature | Class | Shape | Dtype | Description |
---|---|---|---|---|
Translation | ||||
cs | Text | string | ||
en | Text | string |
Supervised keys (See
as_supervised
doc):('cs', 'en')
Examples (tfds.as_dataframe):
wmt16_translate/de-en
Config description: WMT 2016 de-en translation task dataset.
Download size:
1.57 GiB
Dataset size:
1.38 GiB
Auto-cached (documentation): No
Splits:
Split | Examples |
---|---|
'test' |
2,999 |
'train' |
4,548,885 |
'validation' |
2,169 |
- Feature structure:
Translation({
'de': Text(shape=(), dtype=string),
'en': Text(shape=(), dtype=string),
})
- Feature documentation:
Feature | Class | Shape | Dtype | Description |
---|---|---|---|---|
Translation | ||||
de | Text | string | ||
en | Text | string |
Supervised keys (See
as_supervised
doc):('de', 'en')
Examples (tfds.as_dataframe):
wmt16_translate/fi-en
Config description: WMT 2016 fi-en translation task dataset.
Download size:
260.51 MiB
Dataset size:
624.31 MiB
Auto-cached (documentation): No
Splits:
Split | Examples |
---|---|
'test' |
6,000 |
'train' |
2,073,394 |
'validation' |
1,370 |
- Feature structure:
Translation({
'en': Text(shape=(), dtype=string),
'fi': Text(shape=(), dtype=string),
})
- Feature documentation:
Feature | Class | Shape | Dtype | Description |
---|---|---|---|---|
Translation | ||||
en | Text | string | ||
fi | Text | string |
Supervised keys (See
as_supervised
doc):('fi', 'en')
Examples (tfds.as_dataframe):
wmt16_translate/ro-en
Config description: WMT 2016 ro-en translation task dataset.
Download size:
273.83 MiB
Dataset size:
194.35 MiB
Auto-cached (documentation): Yes (test, validation), Only when
shuffle_files=False
(train)Splits:
Split | Examples |
---|---|
'test' |
1,999 |
'train' |
610,320 |
'validation' |
1,999 |
- Feature structure:
Translation({
'en': Text(shape=(), dtype=string),
'ro': Text(shape=(), dtype=string),
})
- Feature documentation:
Feature | Class | Shape | Dtype | Description |
---|---|---|---|---|
Translation | ||||
en | Text | string | ||
ro | Text | string |
Supervised keys (See
as_supervised
doc):('ro', 'en')
Examples (tfds.as_dataframe):
wmt16_translate/ru-en
Config description: WMT 2016 ru-en translation task dataset.
Download size:
993.38 MiB
Dataset size:
854.44 MiB
Auto-cached (documentation): No
Splits:
Split | Examples |
---|---|
'test' |
2,998 |
'train' |
2,516,162 |
'validation' |
2,818 |
- Feature structure:
Translation({
'en': Text(shape=(), dtype=string),
'ru': Text(shape=(), dtype=string),
})
- Feature documentation:
Feature | Class | Shape | Dtype | Description |
---|---|---|---|---|
Translation | ||||
en | Text | string | ||
ru | Text | string |
Supervised keys (See
as_supervised
doc):('ru', 'en')
Examples (tfds.as_dataframe):
wmt16_translate/tr-en
Config description: WMT 2016 tr-en translation task dataset.
Download size:
59.32 MiB
Dataset size:
63.21 MiB
Auto-cached (documentation): Yes
Splits:
Split | Examples |
---|---|
'test' |
3,000 |
'train' |
205,756 |
'validation' |
1,001 |
- Feature structure:
Translation({
'en': Text(shape=(), dtype=string),
'tr': Text(shape=(), dtype=string),
})
- Feature documentation:
Feature | Class | Shape | Dtype | Description |
---|---|---|---|---|
Translation | ||||
en | Text | string | ||
tr | Text | string |
Supervised keys (See
as_supervised
doc):('tr', 'en')
Examples (tfds.as_dataframe):