def METHOD_NAME(
    context: mlrun.MLClientCtx, algorithm_functionality: str, model_name: str = None
):
    algorithm_functionality = AlgorithmFunctionality(algorithm_functionality)
    model = get_model(algorithm_functionality=algorithm_functionality)
    x_train, x_test, y_train, y_test = get_dataset(
        algorithm_functionality=algorithm_functionality, for_training=True
    )
    apply_mlrun(model=model, model_name=model_name, x_test=x_test, y_test=y_test)
    model.fit(x_train, y_train)