hummingbird.ml.operator_converters.sklearn.imputer

Converter for scikit-learn Imputers: SimpleImputer and MissingIndicator

hummingbird.ml.operator_converters.sklearn.imputer.convert_sklearn_missing_indicator(operator, device, extra_config)

Converter for sklearn.impute.MissingIndicator Args:

operator: An operator wrapping a sklearn.impute.MissingIndicator 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.imputer.convert_sklearn_simple_imputer(operator, device, extra_config)

Converter for sklearn.impute.SimpleImputer

Args:

operator: An operator wrapping a sklearn.impute.SimpleImputer 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