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.