code: def get_params():
      """Get parameters from command line"""
      parser = argparse.ArgumentParser()
      parser.add_argument(
          "--data_dir",
          type=str,
          default="/tmp/tensorflow/mnist/input_data",
          help="data directory",
      )
      parser.add_argument("--batch_num", type=int, default=1000)
      parser.add_argument("--batch_size", type=int, default=200)
      args, _ = parser.parse_known_args()
      return args
  
imports: from __future__ import absolute_import, division, print_function
  import logging
  import math
  import tempfile
  import time
  import argparse
  import tensorflow
  from tensorflow.examples.tutorials.mnist import input_data
  import nni
label_desc: Use image node_id 3 for calling a global function with example usage: get_params() and returns: args