pytorch_geometric 源码学习

class Planetoid(InMemoryDataset):
    
    url = ‘https://github.com/kimiyoung/planetoid/raw/master/data‘

    def __init__(self, root, name, transform=None, pre_transform=None):
        self.name = name
        super(Planetoid, self).__init__(root, transform, pre_transform)
        self.data, self.slices = torch.load(self.processed_paths[0])

    @property
    def raw_dir(self):
        return osp.join(self.root, self.name, ‘raw‘)

    @property
    def processed_dir(self):
        return osp.join(self.root, self.name, ‘processed‘)

    @property
    def raw_file_names(self):
        names = [‘x‘, ‘tx‘, ‘allx‘, ‘y‘, ‘ty‘, ‘ally‘, ‘graph‘, ‘test.index‘]
        return [‘ind.{}.{}‘.format(self.name.lower(), name) for name in names]

    @property
    def processed_file_names(self):
        return ‘data.pt‘

    def __repr__(self):
        return ‘{}()‘.format(self.name)