Skip to content

GitLab

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

hotfix (#1987) 


Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-51-214.ec2.internal>
parent 729ff2ef
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
python/dgl/data/reddit.py
View file @ e4b4fcff
...@@ -171,6 +171,9 @@ class RedditDataset(DGLBuiltinDataset): ...@@ -171,6 +171,9 @@ class RedditDataset(DGLBuiltinDataset):
graph_path = os.path.join(self.save_path, 'dgl_graph.bin') graph_path = os.path.join(self.save_path, 'dgl_graph.bin')
graphs, _ = load_graphs(graph_path) graphs, _ = load_graphs(graph_path)
self._graph = graphs[0] self._graph = graphs[0]
self._graph.ndata['train_mask'] = generate_mask_tensor(self._graph.ndata['train_mask'].numpy())
self._graph.ndata['val_mask'] = generate_mask_tensor(self._graph.ndata['val_mask'].numpy())
self._graph.ndata['test_mask'] = generate_mask_tensor(self._graph.ndata['test_mask'].numpy())
self._print_info() self._print_info()
def _print_info(self): def _print_info(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