dialog_re

References:

dialog_re

Use the following command to load this dataset in TFDS:

ds = tfds.load('huggingface:dialog_re/dialog_re')
  • Description:
DialogRE is the first human-annotated dialogue based relation extraction (RE) dataset aiming
to support the prediction of relation(s) between two arguments that appear in a dialogue.
The dataset annotates all occurrences of 36 possible relation types that exist between pairs
of arguments in the 1,788 dialogues originating from the complete transcripts of Friends.
Split Examples
'test' 357
'train' 1073
'validation' 358
  • Features:
{
    "dialog": {
        "feature": {
            "dtype": "string",
            "id": null,
            "_type": "Value"
        },
        "length": -1,
        "id": null,
        "_type": "Sequence"
    },
    "relation_data": {
        "feature": {
            "x": {
                "dtype": "string",
                "id": null,
                "_type": "Value"
            },
            "y": {
                "dtype": "string",
                "id": null,
                "_type": "Value"
            },
            "x_type": {
                "dtype": "string",
                "id": null,
                "_type": "Value"
            },
            "y_type": {
                "dtype": "string",
                "id": null,
                "_type": "Value"
            },
            "r": {
                "feature": {
                    "dtype": "string",
                    "id": null,
                    "_type": "Value"
                },
                "length": -1,
                "id": null,
                "_type": "Sequence"
            },
            "rid": {
                "feature": {
                    "dtype": "int32",
                    "id": null,
                    "_type": "Value"
                },
                "length": -1,
                "id": null,
                "_type": "Sequence"
            },
            "t": {
                "feature": {
                    "dtype": "string",
                    "id": null,
                    "_type": "Value"
                },
                "length": -1,
                "id": null,
                "_type": "Sequence"
            }
        },
        "length": -1,
        "id": null,
        "_type": "Sequence"
    }
}