Skip to content
Permalink
Browse files

add shuffling

  • Loading branch information...
rusty1s committed Nov 5, 2019
1 parent 6b0a84c commit 40344b7e6e76be1bf7e61df97e07f776c5535a2e
Showing with 1 addition and 1 deletion.
  1. +1 −1 examples/qm9_nn_conv.py
@@ -49,7 +49,7 @@ def __call__(self, data):

path = osp.join(osp.dirname(osp.realpath(__file__)), '..', 'data', 'QM9')
transform = T.Compose([MyTransform(), Complete(), T.Distance(norm=False)])
dataset = QM9(path, transform=transform)
dataset = QM9(path, transform=transform).shuffle()

# Normalize targets to mean = 0 and std = 1.
mean = dataset.data.y.mean(dim=0, keepdim=True)

0 comments on commit 40344b7

Please sign in to comment.
You can’t perform that action at this time.