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])