code: def load(name):
      if name == "cora":
          dataset = pgl.dataset.CoraDataset()
      elif name == "pubmed":
          dataset = pgl.dataset.CitationDataset(
              "pubmed", symmetry_edges=True
          )
      elif name == "citeseer":
          dataset = pgl.dataset.CitationDataset(
              "citeseer", symmetry_edges=True
          )
      else:
          raise ValueError(name + " dataset doesn't exists")
  
      return dataset
  
file_path: PGL/examples/AITISA/load_model_from_pyg/gcn.py
imports: import pgl
  import paddle
  import torch
  import torch.nn.functional
  from torch_geometric.nn import GCNConv
  import argparse
  import numpy
label_desc: Use image node_id 1 for calling a global function with example usage: load(name) and returns: dataset