datasets.kg_data

Module Contents

Functions

read_triplet_data(folder)

datasets.kg_data.read_triplet_data(folder)[source]
class datasets.kg_data.KnowledgeGraphDataset(root, name)[source]

Bases: cogdl.data.Dataset

url = https://raw.githubusercontent.com/thunlp/OpenKE/OpenKE-PyTorch/benchmarks[source]
property raw_file_names(self)[source]
property processed_file_names(self)[source]
get(self, idx)[source]
download(self)[source]
process(self)[source]
class datasets.kg_data.FB13Datset[source]

Bases: datasets.kg_data.KnowledgeGraphDataset

class datasets.kg_data.FB15kDatset[source]

Bases: datasets.kg_data.KnowledgeGraphDataset

class datasets.kg_data.FB15k237Datset[source]

Bases: datasets.kg_data.KnowledgeGraphDataset

class datasets.kg_data.WN18Datset[source]

Bases: datasets.kg_data.KnowledgeGraphDataset

class datasets.kg_data.WN18RRDataset[source]

Bases: datasets.kg_data.KnowledgeGraphDataset