code: def load(name):
      if name == "BlogCatalog":
          dataset = data_loader.BlogCatalogDataset()
      elif name == "ArXiv":
          dataset = data_loader.ArXivDataset()
      else:
          raise ValueError(name + " dataset doesn't exists")
      return dataset
  
file_path: PGL/legacy/examples/deepwalk/link_predict.py
imports: import argparse
  import time
  import math
  import os
  import numpy
  from sklearn import metrics
  import pgl
  from pgl import data_loader
  from pgl.utils import op
  from pgl.utils.logger import log
  import paddle.fluid
  import paddle.fluid.layers
label_desc: Use image node_id 1 for calling a global function with example usage: load(name) and returns: dataset