hummingbird.ml.operator_converters.onnx.tree_ensemble

Converters for ONNX-ML tree-ensemble models.

hummingbird.ml.operator_converters.onnx.tree_ensemble._dummy_get_parameter(tree_info, extra_config)

Dummy function used to return parameters (TreeEnsemble converters already have parameters in the right format)

hummingbird.ml.operator_converters.onnx.tree_ensemble._get_tree_infos_from_onnx_ml_operator(model)

Function used to extract the parameters from a ONNXML TreeEnsemble model.

hummingbird.ml.operator_converters.onnx.tree_ensemble._get_tree_infos_from_tree_ensemble(operator, device=None, extra_config={})

Base method for extracting parameters from `ai.onnx.ml.TreeEnsemble`s.

hummingbird.ml.operator_converters.onnx.tree_ensemble.convert_onnx_tree_ensemble_classifier(operator, device=None, extra_config={})

Converter for ai.onnx.ml.TreeEnsembleClassifier.

Args:

operator: An operator wrapping a ai.onnx.ml.TreeEnsembleClassifier 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.onnx.tree_ensemble.convert_onnx_tree_ensemble_regressor(operator, device=None, extra_config={})

Converter for ai.onnx.ml.TreeEnsembleRegressor.

Args:

operator: An operator wrapping a ai.onnx.ml.TreeEnsembleRegressor 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