def neo_postprocess()

in mxnet_managed_spot_training_checkpointing/source_dir/mnist.py [0:0]

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

def neo_postprocess(result):
    import logging
    import numpy as np
    import json

    logging.info('Invoking user-defined post-processing function')

    # Softmax (assumes batch size 1)
    result = np.squeeze(result)
    result_exp = np.exp(result - np.max(result))
    result = result_exp / np.sum(result_exp)

    response_body = json.dumps(result.tolist())
    content_type = 'application/json'

    return response_body, content_type