kiez.io.data_loading

Convenience methods for loading entity embeddings from knowledge graph embeddings.

Functions

from_openea(emb_dir_path, kg_path)

Load OpenEA-type data.

kiez.io.data_loading.from_openea(emb_dir_path: str, kg_path: str) → Tuple[ndarray, ndarray, Dict[int, str], Dict[int, str], Dict[str, str]]

Load OpenEA-type data.

Parameters:

• emb_dir_path (str) – Path to folder containing embedding

• kg_path (str) – Path to folder containing kg info

Returns:

embeddings of both knowledge graphs entity ids per embedding row for both knowledge graphs entity links

Return type:

emb1, emb2, kg1_ids_new, kg2_ids_new, ent_links_new

Notes

See here for more information on the dataset structure https://github.com/nju-websoft/OpenEA#dataset-description