code: def init_data():
      train = h2o.import_file(TRAIN_DATASET)
      test = h2o.import_file(TEST_DATASET)
  
      return {
          "predictors": train.columns,
          "response": "species",
          "train": train,
          "test": test,
      }
  
imports: import h2o
  from tests import pyunit_utils
  from h2o.estimators.xgboost import H2OXGBoostEstimator
  from h2o.exceptions import H2OResponseError, H2OValueError
  from h2o.grid.grid_search import H2OGridSearch
label_desc: Use image node_id 1 for calling a global function with example usage: init_data() and returns: dict