614393f8ac305d38ed65dda084d86f45a1b465f5,torch_geometric/dataset/qm9.py,QM9,process,#QM9#,43

Before Change

        // Add targets to dataset.
        with open(self._raw_files[1], "r") as f:
            src = [x.split(",")[4:16] for x in f.read().split("\n")[1:-1]]
            y = torch.Tensor([[float(y) for y in x] for x in src])
            dataset.y = y
            slices["y"] = torch.arange(y.size(0) + 1, out=torch.LongTensor())

After Change

        // Remove invalid data.
        with open(self._raw_files[2], "r") as f:
            src = split2d(f.read().split("\n")[9:-2], end=1, out=LongTensor())
            split = torch.arange(dataset.y.size(0), out=LongTensor())
            split = split[split == split.clone().index_fill_(0, src, -1)]

Instances