def model_fn()

in entrypoint.py [0:0]

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

def model_fn(model_dir):
    """Deserialize and return fitted model.

    Note that this should have the same name as the serialized model in the _xgb_train method
    """
    model_file = 'xgboost-model'
    booster = pkl.load(open(os.path.join(model_dir, model_file), 'rb'))
    return booster