def __init__(self, root, name):
    self.name = name
    self.url = f'https://github.com/cenyk1230/gtn-data/blob/master/{name}.zip?raw=true'
    super(GTNDataset, self).__init__(root)
    self.data = torch.load(self.processed_paths[0])
    self.num_classes = (torch.max(self.data.train_target).item() + 1)
    self.num_edge = len(self.data.adj)
    self.num_nodes = self.data.x.shape[0]
def __init__(self, root, name):
    self.name = name
    self.url = f'https://github.com/cenyk1230/gtn-data/blob/master/{name}.zip?raw=true'
    super(GTNDataset, self).__init__(root)
    self.data = torch.load(self.processed_paths[0])
    self.num_classes = (torch.max(self.data.train_target).item() + 1)
    self.num_edge = len(self.data.adj)
    self.num_nodes = self.data.x.shape[0]