hummingbird.ml.operator_converters.sklearn.gbdt¶
Converters for Sklearn’s GradientBoosting models.
- hummingbird.ml.operator_converters.sklearn.gbdt._get_parameters_hist_gbdt(trees, extra_config)[source]¶
Extract the tree parameters from SklearnHistGradientBoostingClassifier trees Args:
trees: The information representing a tree (ensemble) Returns: The tree parameters wrapped into an instance of operator_converters._tree_commons_TreeParameters
- hummingbird.ml.operator_converters.sklearn.gbdt.convert_sklearn_gbdt_classifier(operator, device, extra_config)[source]¶
Converter for sklearn.ensemble.GradientBoostingClassifier
- Args:
- operator: An operator wrapping a sklearn.ensemble.GradientBoostingClassifier
or sklearn.ensemble.HistGradientBoostingClassifier model
device: String defining the type of device the converted operator should be run on extra_config: Extra configuration used to select the best conversion strategy
- Returns:
A PyTorch model
- hummingbird.ml.operator_converters.sklearn.gbdt.convert_sklearn_gbdt_regressor(operator, device, extra_config)[source]¶
Converter for sklearn.ensemble.GradientBoostingRegressor.
- Args:
operator: An operator wrapping a sklearn.ensemble.GradientBoostingRegressor or sklearn.ensemble.HistGradientBoostingRegressor model device: String defining the type of device the converted operator should be run on extra_config: Extra configuration used to select the best conversion strategy
- Returns:
A PyTorch model
- hummingbird.ml.operator_converters.sklearn.gbdt.convert_sklearn_hist_gbdt_classifier(operator, device, extra_config)[source]¶
Converter for sklearn.ensemble.HistGradientBoostingClassifier
- Args:
operator: An operator wrapping a sklearn.ensemble.HistGradientBoostingClassifier model device: String defining the type of device the converted operator should be run on extra_config: Extra configuration used to select the best conversion strategy
- Returns:
A PyTorch model
- hummingbird.ml.operator_converters.sklearn.gbdt.convert_sklearn_hist_gbdt_regressor(operator, device, extra_config)[source]¶
Converter for sklearn.ensemble.HistGradientBoostingRegressor
- Args:
operator: An operator wrapping a sklearn.ensemble.HistGradientBoostingRegressor model device: String defining the type of device the converted operator should be run on extra_config: Extra configuration used to select the best conversion strategy
- Returns:
A PyTorch model