Python split_dataset_general Examples

Example #1

    def split_dataset(self, dataset, args):
        # process each graph and add it into Data() as attribute tx
        for i, data in enumerate(dataset):
            new_feature = get_single_feature(dataset[i], args.num_features,
                                             args.num_classes, args.sample,
                                             args.neighbor, args.stride)
            dataset[i].tx = torch.from_numpy(new_feature)

        return split_dataset_general(dataset, args)

Example #2

File: diffpool.py Project: jkx19/cogdl

 def split_dataset(cls, dataset, args):
     return split_dataset_general(dataset, args)