Source code for pykeen.datasets.wd50k

# -*- coding: utf-8 -*-

"""The Wikidata-based WD50K dataset (triple-only version) from [galkin2020]_.

- GitHub Repository:
- Paper:

Get a summary with ``python -m pykeen.datasets.wd50k``,

import click
from docdata import parse_docdata
from more_click import verbose_option

from .base import UnpackedRemoteDataset

TRIPLES_VALID_URL = f"{BASE_URL}/triples/valid.txt"
TRIPLES_TEST_URL = f"{BASE_URL}/triples/test.txt"
TRIPLES_TRAIN_URL = f"{BASE_URL}/triples/train.txt"

[docs]@parse_docdata class WD50KT(UnpackedRemoteDataset): """The triples-only version of WD50K. --- name: WD50K (triples) citation: author: Galkin year: 2020 link: arxiv: 2009.10847 github: migalkin/StarE statistics: entities: 40107 relations: 473 training: 164631 testing: 45284 validation: 22429 triples: 232344 """ def __init__(self, create_inverse_triples: bool = False, **kwargs): """Initialize the WD50K (triples) dataset from [galkin2020]_. :param create_inverse_triples: Should inverse triples be created? Defaults to false. :param kwargs: keyword arguments passed to :class:`pykeen.datasets.base.UnpackedRemoteDataset`. """ super().__init__( training_url=TRIPLES_TRAIN_URL, testing_url=TRIPLES_TEST_URL, validation_url=TRIPLES_VALID_URL, create_inverse_triples=create_inverse_triples, load_triples_kwargs={"delimiter": ","}, **kwargs, )
@click.command() @verbose_option def _main(): for cls in [WD50KT]: click.secho(f"Loading {cls.__name__}", fg="green", bold=True) d = cls() d.summarize() if __name__ == "__main__": _main()