Source code for pykeen.datasets.kinships

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

"""Get triples from the Kinships dataset."""

import pathlib

from docdata import parse_docdata

from ..base import PathDataset

__all__ = [

HERE = pathlib.Path(__file__).resolve().parent

KINSHIPS_TRAIN_PATH = HERE.joinpath("train.txt")
KINSHIPS_TEST_PATH = HERE.joinpath("test.txt")
KINSHIPS_VALIDATE_PATH = HERE.joinpath("valid.txt")

[docs]@parse_docdata class Kinships(PathDataset): """The Kinships dataset. --- name: Kinships statistics: entities: 104 relations: 25 training: 8544 testing: 1074 validation: 1068 triples: 10686 citation: author: Kemp year: 2006 link: """ def __init__(self, create_inverse_triples: bool = False, **kwargs): """Initialize the Kinships dataset. :param create_inverse_triples: Should inverse triples be created? Defaults to false. :param kwargs: keyword arguments passed to :class:`pykeen.datasets.base.PathDataset`. """ super().__init__( training_path=KINSHIPS_TRAIN_PATH, testing_path=KINSHIPS_TEST_PATH, validation_path=KINSHIPS_VALIDATE_PATH, create_inverse_triples=create_inverse_triples, **kwargs, )
if __name__ == "__main__": Kinships().summarize()