Source code for experimental_experiment.xoptim.patterns_ml

from typing import List




def get_ml_patterns(
    verbose: int = 0,
) -> List["PatternOptimization"]:  # noqa: F821
    """
    Returns a default list of optimization patterns for ai.onnx.ml.
    It is equal to the following list.

    .. runpython::
        :showcode:

        import pprint
        from experimental_experiment.xoptim.patterns_ml import (
            get_ml_patterns,
        )
        pprint.pprint(get_ml_patterns())
    """
    from .tree_ensemble import (
        TreeEnsembleRegressorConcatPattern,
        TreeEnsembleRegressorMulPattern,
    )

    return [
        TreeEnsembleRegressorConcatPattern(verbose=verbose),
        TreeEnsembleRegressorMulPattern(verbose=verbose),
    ]