salient_span_wikipedia

  • Description:

Wikipedia sentences with labeled salient spans.

@article{guu2020realm,
    title={REALM: Retrieval-Augmented Language Model Pre-Training},
    author={Kelvin Guu and Kenton Lee and Zora Tung and Panupong Pasupat and Ming-Wei Chang},
    year={2020},
    journal = {arXiv e-prints},
    archivePrefix = {arXiv},
    eprint={2002.08909},
}

salient_span_wikipedia/sentences (default config)

  • Config description: Examples are individual sentences containing entities.

  • Dataset size: 20.57 GiB

  • Splits:

Split Examples
'train' 82,291,706
  • Features:
FeaturesDict({
    'spans': Sequence({
        'limit': tf.int32,
        'start': tf.int32,
        'type': tf.string,
    }),
    'text': Text(shape=(), dtype=tf.string),
    'title': Text(shape=(), dtype=tf.string),
})

salient_span_wikipedia/documents

  • Config description: Examples re full documents.

  • Dataset size: 16.52 GiB

  • Splits:

Split Examples
'train' 13,353,718
  • Features:
FeaturesDict({
    'sentences': Sequence({
        'limit': tf.int32,
        'start': tf.int32,
    }),
    'spans': Sequence({
        'limit': tf.int32,
        'start': tf.int32,
        'type': tf.string,
    }),
    'text': Text(shape=(), dtype=tf.string),
    'title': Text(shape=(), dtype=tf.string),
})