def model_fn()

in code/scikit_learn_iris.py [0:0]

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

def model_fn(model_dir):
    """Deserialized and return fitted model
    Note that this should have the same name as the serialized model in the main method
    """
    clf = load(os.path.join(model_dir, "model.joblib"))
    return clf