def save_model()

in source_directory/training/training_script.py [0:0]

• 5 lines of code
• 1 McCabe index (conditional complexity)

def save_model(model, model_output_dir):
    
    # https://www.tensorflow.org/guide/keras/save_and_serialize
    logging.info("Saving model used model.save()")

    tf.saved_model.save(model, os.path.join(model_output_dir, 'cinic10_classifier', '1'))

    logging.info("Model successfully saved to: {}".format(model_output_dir))

    return