code: def load_dataset(dataset_name):
      if dataset_name == "higgs":
          return higgs()
      elif dataset_name == "log1p":
          return log1p()
      elif dataset_name == "cifar":
          return cifar()
      elif dataset_name == "news20":
          return news20()
  
file_path: thundergbm/python/benchmarks/experiments.py
imports: import utils.data_utils
  from model.catboost_model import CatboostModel
  from model.lightgbm_model import LightGBMModel
  from model.xgboost_model import XGboostModel
  from model.thundergbm_model import ThunderGBMModel
  from model.datasets import Dataset
  import utils.file_utils
  import pandas
  import math
  import sys
label_desc: Use image node_id 7 for calling a global function with example usage: load_dataset(dataset_name) and returns: higgs, log1p, cifar, news20