Source code for pykeen.datasets.globi

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

"""The Global Biotic Interactions (GloBI) dataset.

Get a summary with ``python -m pykeen.datasets.globi``

import click
from docdata import parse_docdata
from more_click import verbose_option

from .base import SingleTabbedDataset
from ..typing import TorchRandomHint

__all__ = [

URL = ""

[docs]@parse_docdata class Globi(SingleTabbedDataset): """The Global Biotic Interactions (GloBI) dataset. --- name: Global Biotic Interactions citation: author: Poelen year: 2014 link: single: true statistics: entities: 404207 relations: 39 triples: 1966385 training: 1573108 testing: 196638 validation: 196639 """ def __init__( self, random_state: TorchRandomHint = 0, **kwargs, ): """Initialize the GloBI dataset. :param random_state: The random seed to use in splitting the dataset. Defaults to 0. :param kwargs: keyword arguments passed to :class:`pykeen.datasets.base.SingleTabbedDataset`. """ super().__init__( url=URL, random_state=random_state, read_csv_kwargs=dict( usecols=["sourceTaxonId", "interactionTypeName", "targetTaxonId"], ), **kwargs, )
@click.command() @verbose_option def _main(): from pykeen.datasets import get_dataset ds = get_dataset(dataset=Globi) ds.summarize() if __name__ == "__main__": _main()