Skip to content

GitLab

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

hotfix (#1980) 


Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-87-240.ec2.internal>
parent b9ef70e5
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
python/dgl/data/knowledge_graph.py
View file @ e7515773
...@@ -166,9 +166,9 @@ class KnowledgeGraphDataset(DGLBuiltinDataset): ...@@ -166,9 +166,9 @@ class KnowledgeGraphDataset(DGLBuiltinDataset):
if self.verbose: if self.verbose:
print("# entities: {}".format(self.num_nodes)) print("# entities: {}".format(self.num_nodes))
print("# relations: {}".format(self.num_rels)) print("# relations: {}".format(self.num_rels))
print("# training edges: {}".format(train_idx.shape[0])) print("# training edges: {}".format(self._train.shape[0]))
print("# validation edges: {}".format(valid_idx.shape[0])) print("# validation edges: {}".format(self._valid.shape[0]))
print("# testing edges: {}".format(test_idx.shape[0])) print("# testing edges: {}".format(self._test.shape[0]))
@property @property
def num_nodes(self): def num_nodes(self):
......
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