code: def get_model(config):
      input_data = mx.symbol.Variable("input")
      y_true = mx.symbol.Variable("label")
      fc1 = mx.symbol.FullyConnected(
          data=input_data, num_hidden=20, name="fc1"
      )
      fc2 = mx.symbol.FullyConnected(
          data=fc1, num_hidden=10, name="fc2"
      )
      output = mx.symbol.SoftmaxOutput(
          data=fc2, label=y_true, name="output"
      )
      mod = mx.mod.Module(
          symbol=output,
          data_names=["input"],
          label_names=["label"],
          context=mx.cpu(),
      )
      return mod
  
file_path: analytics-zoo/pyzoo/test/zoo/orca/learn/ray/mxnet/test_mxnet_symbol.py
imports: from unittest import TestCase
  import numpy
  import pytest
  import mxnet
  from zoo.orca.learn.mxnet import Estimator, create_config
label_desc: Use image node_id 3 for calling a global function with example usage: get_model(config) and returns: mod