Module hummingbird.ml.operator_converters.sklearn

All scikit-learn operators converters are stored under this package.

Expand source code Browse git
# -------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for
# license information.
# --------------------------------------------------------------------------

"""
All scikit-learn operators converters are stored under this package.
"""

Sub-modules

hummingbird.ml.operator_converters.sklearn.lightgbm

Converters for LightGBM models.

hummingbird.ml.operator_converters.sklearn.skl_array_feature_extractor

Converters for scikit-learn feature selectors: SelectKBest, SelectPercentile, VarianceThreshold.

hummingbird.ml.operator_converters.sklearn.skl_decision_tree

Converters for scikit-learn decision-tree-based models: DecisionTree, RandomForest and ExtraTrees.

hummingbird.ml.operator_converters.sklearn.skl_gbdt

Converters for Sklearn's GradientBoosting models.

hummingbird.ml.operator_converters.sklearn.skl_iforest

Converters for scikit-learn isolation forest.

hummingbird.ml.operator_converters.sklearn.skl_linear

Converters for scikit-learn linear models: LinearRegression, LogisticRegression, LinearSVC, SGDClassifier, LogisticRegressionCV.

hummingbird.ml.operator_converters.sklearn.skl_normalizer

Converter for scikit-learn Normalizer.

hummingbird.ml.operator_converters.sklearn.skl_one_hot_encoder

Converter for scikit-learn one hot encoder.

hummingbird.ml.operator_converters.sklearn.skl_scaler

Converters for scikit-learn scalers: RobustScaler, MaxAbsScaler, MinMaxScaler, StandardScaler.

hummingbird.ml.operator_converters.sklearn.skl_sv

Converters for scikit-learn SV models: SVC, NuSVC. (LinearSVC is covered by linear_classifier.py).

hummingbird.ml.operator_converters.sklearn.xgb

Converters for XGBoost models.