References:
Use the following command to load this dataset in TFDS:
ds = tfds.load('huggingface:docred')
- Description:
Multiple entities in a document generally exhibit complex inter-sentence relations, and cannot be well handled by existing relation extraction (RE) methods that typically focus on extracting intra-sentence relations for single entity pairs. In order to accelerate the research on document-level RE, we introduce DocRED, a new dataset constructed from Wikipedia and Wikidata with three features:
- DocRED annotates both named entities and relations, and is the largest human-annotated dataset for document-level RE from plain text.
- DocRED requires reading multiple sentences in a document to extract entities and infer their relations by synthesizing all information of the document.
- Along with the human-annotated data, we also offer large-scale distantly supervised data, which enables DocRED to be adopted for both supervised and weakly supervised scenarios.
- License: No known license
- Version: 0.0.0
- Splits:
Split | Examples |
---|---|
'test' |
1000 |
'train_annotated' |
3053 |
'train_distant' |
101873 |
'validation' |
998 |
- Features:
{
"title": {
"dtype": "string",
"id": null,
"_type": "Value"
},
"sents": {
"feature": {
"feature": {
"dtype": "string",
"id": null,
"_type": "Value"
},
"length": -1,
"id": null,
"_type": "Sequence"
},
"length": -1,
"id": null,
"_type": "Sequence"
},
"vertexSet": [
[
{
"name": {
"dtype": "string",
"id": null,
"_type": "Value"
},
"sent_id": {
"dtype": "int32",
"id": null,
"_type": "Value"
},
"pos": {
"feature": {
"dtype": "int32",
"id": null,
"_type": "Value"
},
"length": -1,
"id": null,
"_type": "Sequence"
},
"type": {
"dtype": "string",
"id": null,
"_type": "Value"
}
}
]
],
"labels": {
"feature": {
"head": {
"dtype": "int32",
"id": null,
"_type": "Value"
},
"tail": {
"dtype": "int32",
"id": null,
"_type": "Value"
},
"relation_id": {
"dtype": "string",
"id": null,
"_type": "Value"
},
"relation_text": {
"dtype": "string",
"id": null,
"_type": "Value"
},
"evidence": {
"feature": {
"dtype": "int32",
"id": null,
"_type": "Value"
},
"length": -1,
"id": null,
"_type": "Sequence"
}
},
"length": -1,
"id": null,
"_type": "Sequence"
}
}