Skip to content

GitLab

Unverified Commit 4464b973 authored by xiang song(charlie.song)'s avatar xiang song(charlie.song) Committed by GitHub
Browse files

Fix kg dataset len() (#2475) 


Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-59-204.ec2.internal>
parent 4208ce2b
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
python/dgl/data/knowledge_graph.py
View file @ 4464b973
......@@ -464,7 +464,7 @@ class FB15k237Dataset(KnowledgeGraphDataset):
def __len__(self):
r"""The number of graphs in the dataset."""
return super(FB15k237Dataset, self).__len__(idx)
return super(FB15k237Dataset, self).__len__()
class FB15kDataset(KnowledgeGraphDataset):
r"""FB15k link prediction dataset.
......@@ -601,7 +601,7 @@ class FB15kDataset(KnowledgeGraphDataset):
def __len__(self):
r"""The number of graphs in the dataset."""
return super(FB15kDataset, self).__len__(idx)
return super(FB15kDataset, self).__len__()
class WN18Dataset(KnowledgeGraphDataset):
r""" WN18 link prediction dataset.
......@@ -737,7 +737,7 @@ class WN18Dataset(KnowledgeGraphDataset):
def __len__(self):
r"""The number of graphs in the dataset."""
return super(WN18Dataset, self).__len__(idx)
return super(WN18Dataset, self).__len__()
def load_data(dataset):
r"""Load knowledge graph dataset for RGCN link prediction tasks
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment