code: def _get_test_data(in_size=32):
      data = np.ones((in_size, 100))
      label = np.zeros((in_size, 1))
      data_arr = mx.gluon.data.dataset.ArrayDataset(data, label)
      return mx.gluon.data.DataLoader(data_arr, batch_size=8)
  
imports: import os
  import logging
  import sys
  import re
  import mxnet
  from common import TemporaryDirectory
  from mxnet import np
  from mxnet.gluon import nn, loss
  from mxnet.gluon.contrib.estimator import estimator, event_handler
  from mxnet.gluon.contrib.estimator.event_handler import LoggingHandler
  from mxnet.gluon.contrib.estimator.event_handler import ValidationHandler
  from mxnet.gluon.contrib.estimator import EpochEnd
  from mxnet.gluon.data.dataset import Dataset
label_desc: Use image node_id 2 for calling a global function with example usage: _get_test_data(in_size) and returns: mx