6b0a84c6977fc79bb2953d55e4d36db530f35496,examples/qm9_nn_conv.py,,,#,50

Before Change

dataset = QM9(path, transform=transform).shuffle()

// Normalize targets to mean = 0 and std = 1.
mean = dataset.data.y[:, target].mean().item()
std = dataset.data.y[:, target].std().item()
dataset.data.y[:, target] = (dataset.data.y[:, target] - mean) / std

// Split datasets.

After Change

dataset = QM9(path, transform=transform)

// Normalize targets to mean = 0 and std = 1.
mean = dataset.data.y.mean(dim=0, keepdim=True)
std = dataset.data.y.std(dim=0, keepdim=True)
dataset.data.y = (dataset.data.y - mean) / std
mean, std = mean[:, target].item(), std[:, target].item()

// Split datasets.
test_dataset = dataset[:10000]
val_dataset = dataset[10000:20000]

Instances