def model_fn()

in autogluon-tab-with-test.py [0:0]


def model_fn(model_dir):
  """
  Load the gluon model. Called once when hosting service starts.
  :param: model_dir The directory where model files are stored.
  :return: a model (in this case an AutoGluon network)
  """
  net = task.load(model_dir)
  return net