def process(self):
if self.name.lower() == "cora":
data = read_npz(self.raw_paths[0])
data = data if self.pre_transform is None else self.pre_transform(
data)
data, slices = collate_and_pca(self, [data], pca_dim=self.pca_dim)
torch.save((data, slices), self.processed_paths[0])
else:
return super().process()
def process(self):
data = read_npz(self.raw_paths[0])
data = data if self.pre_transform is None else self.pre_transform(data)
data, slices = self.collate([data])
torch.save((data, slices), self.processed_paths[0])
def process(self):
data = read_npz(self.raw_paths[0])
data = data if self.pre_transform is None else self.pre_transform(data)
data, slices = collate_and_pca(self, [data], pca_dim=self.pca_dim)
torch.save((data, slices), self.processed_paths[0])