Module hummingbird.ml.operator_converters
All operators converters are stored under this package.
Expand source code
# -------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for
# license information.
# --------------------------------------------------------------------------
"""
All operators converters are stored under this package.
"""
# Register constants used within Hummingbird converters.
from . import constants as converter_constants
from .. import supported as hummingbird_constants
from .._utils import _Constants
# Add constants in scope.
constants = _Constants(converter_constants, hummingbird_constants)
# To register a converter for scikit-learn API operators, import associated modules here.
from . import lightgbm # noqa: E402
from . import onnx_operator # noqa: E402
from . import onnxml_array_feature_extractor # noqa: E402
from . import onnxml_linear # noqa: E402
from . import onnxml_normalizer # noqa: E402
from . import onnxml_scaler # noqa: E402
from . import onnxml_tree_ensemble # noqa: E402
from . import skl_array_feature_extractor # noqa: E402
from . import skl_decision_tree # noqa: E402
from . import skl_gbdt # noqa: E402
from . import skl_iforest # noqa: E402
from . import skl_linear # noqa: E402
from . import skl_normalizer # noqa: E402
from . import skl_one_hot_encoder # noqa: E402
from . import skl_scaler # noqa: E402
from . import skl_sv # noqa: E402
from . import xgb # noqa: E402
__pdoc__ = {}
__pdoc__["hummingbird.operator_converters._array_feature_extractor_implementations"] = True
__pdoc__["hummingbird.operator_converters._gbdt_commons"] = True
__pdoc__["hummingbird.operator_converters._tree_commons"] = True
__pdoc__["hummingbird.operator_converters._linear_implementations"] = True
__pdoc__["hummingbird.operator_converters._normalizer_implementations"] = True
__pdoc__["hummingbird.operator_converters._scaler_implementations"] = True
__pdoc__["hummingbird.operator_converters._tree_implementations"] = True
Sub-modules
hummingbird.ml.operator_converters.constants
-
Constants used in the Hummingbird converters are defined here.
hummingbird.ml.operator_converters.lightgbm
-
Converters for LightGBM models.
hummingbird.ml.operator_converters.onnx_operator
-
Converters for ONNX operators.
hummingbird.ml.operator_converters.onnxml_array_feature_extractor
hummingbird.ml.operator_converters.onnxml_linear
-
Converters for ONNX-ML linear models.
hummingbird.ml.operator_converters.onnxml_normalizer
hummingbird.ml.operator_converters.onnxml_scaler
hummingbird.ml.operator_converters.onnxml_tree_ensemble
-
Converters for ONNX-ML tree-ensemble models.
hummingbird.ml.operator_converters.skl_array_feature_extractor
-
Converters for scikit-learn feature selectors: SelectKBest, SelectPercentile, VarianceThreshold
hummingbird.ml.operator_converters.skl_decision_tree
-
Converters for scikit-learn decision-tree-based models: DecisionTree, RandomForest and ExtraTrees.
hummingbird.ml.operator_converters.skl_gbdt
-
Converters for Sklearn's GradientBoosting models.
hummingbird.ml.operator_converters.skl_iforest
-
Converters for scikit-learn isolation forest.
hummingbird.ml.operator_converters.skl_linear
hummingbird.ml.operator_converters.skl_normalizer
hummingbird.ml.operator_converters.skl_one_hot_encoder
hummingbird.ml.operator_converters.skl_scaler
hummingbird.ml.operator_converters.skl_sv
hummingbird.ml.operator_converters.xgb
-
Converters for XGBoost models.