class_docstr: The OpenEA dataset family.
  
  ---
  name: OpenEA Family
  citation:
      author: Sun
      year: 2020
      link: http://www.vldb.org/pvldb/vol13/p2326-sun.pdf
  single: true
  statistics:
      entities: 15000
      relations: 248
      triples: 38265
      training: 30612
      testing: 3826
      validation: 3827
class_docstr_tok: ["The","OpenEA","dataset","family",".","--","-","name",":","OpenEA","Family","citation",":","author",":","Sun","year",":","2020","link",":","http",":","\/\/www.vldb.org\/pvldb\/vol13\/p2326-sun.pdf","single",":","true","statistics",":","entities",":","15000","relations",":","248","triples",":","38265","training",":","30612","testing",":","3826","validation",":","3827"]
docstr: Initialize the dataset.
  
  :param graph_pair:
      The graph-pair within the dataset family (cf. GRAPH_PAIRS).
  :param size:
      The size of the graphs (either "15K" or "100K").
  :param version:
      The version of the pairing (either "V1" or "V2). "V1" has lower connectivity in the graph than "V2".
  :param kwargs:
      additional keyword-based parameters passed to :meth:`EABase.__init__`
  
  :raises ValueError:
      If the graph pair, size or version is invalid.
docstr_tok: ["Initialize","the","dataset",".",":","param","graph_pair",":","The","graph-pair","within","the","dataset","family","(","cf",".","GRAPH_PAIRS",")",".",":","param","size",":","The","size","of","the","graphs","(","either","``","15K","''","or","``","100K","''",")",".",":","param","version",":","The","version","of","the","pairing","(","either","``","V1","''","or","``","V2",")",".","``","V1","''","has","lower","connectivity","in","the","graph","than","``","V2","''",".",":","param","kwargs",":","additional","keyword-based","parameters","passed","to",":","meth",":","`","EABase.__init__","`",":","raises","ValueError",":","If","the","graph","pair",",","size","or","version","is","invalid","."]
code: def __init__(
      self,
      *,
      graph_pair: str = D_W,
      size: GraphSize = SIZE_15K,
      version: GraphVersion = V1,
      **kwargs,
  ):
      """
      Initialize the dataset.
  
      :param graph_pair:
          The graph-pair within the dataset family (cf. GRAPH_PAIRS).
      :param size:
          The size of the graphs (either "15K" or "100K").
      :param version:
          The version of the pairing (either "V1" or "V2). "V1" has lower connectivity in the graph than "V2".
      :param kwargs:
          additional keyword-based parameters passed to :meth:`EABase.__init__`
  
      :raises ValueError:
          If the graph pair, size or version is invalid.
      """
      # Input validation.
      if graph_pair not in GRAPH_PAIRS:
          raise ValueError(
              f"Invalid graph pair: Allowed are: {GRAPH_PAIRS}"
          )
      if size not in GRAPH_SIZES:
          raise ValueError(f"size must be one of {GRAPH_SIZES}")
      if version not in GRAPH_VERSIONS:
          raise ValueError(f"version must be one of {GRAPH_VERSIONS}")
  
      # ensure zip file is present
      self.zip_path = OPEN_EA_MODULE.ensure(
          url=OpenEA.FIGSHARE_LINK,
          name="OpenEA_dataset_v2.0.zip",
          download_kwargs=dict(hexdigests=dict(sha512=OpenEA.SHA512)),
      )
      # save relative paths beforehand so they are present for loading
      self.inner_path = pathlib.PurePosixPath(
          "OpenEA_dataset_v2.0", f"{graph_pair}_{size}_{version}"
      )
      # delegate to super class
      super().__init__(**kwargs)
  
file_path: pykeen/src/pykeen/datasets/ea/openea.py
imports: import itertools
  import logging
  import pathlib
  from typing import Literal, Tuple
  import click
  import pandas
  from docdata import parse_docdata
  from more_click import verbose_option
  from pystow.utils import read_zipfile_csv
  from .base import EADataset
  from ...constants import COLUMN_LABELS, PYKEEN_DATASETS_MODULE
  from ...triples import TriplesFactory
  from ...typing import EA_SIDE_LEFT, EA_SIDES, EASide
label_desc: Use image node_id 1 to create a new OpenEA object from inherited base classes: EADataset with example: obj = OpenEA()