def export_model()

in china/2020_GCR_Kubeflow_Workshop/resources/mnist-tensorflow-jupyter.py [0:0]

• 9 lines of code
• 2 McCabe index (conditional complexity)

def export_model(model, model_export_path):
  version = 1
  export_path = os.path.join(model_export_path, str(version))

  tf.saved_model.simple_save(
    keras.backend.get_session(),
    export_path,
    inputs={'input_image': model.input},
    outputs={t.name:t for t in model.outputs})

  print('\nSaved model: {}'.format(export_path))