def model_fn()

in code/train.py [0:0]

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

def model_fn(model_dir):
    predictor = joblib.load(os.path.join(model_dir, "model.joblib"))
    return predictor