hummingbird.ml.operator_converters.sklearn.one_hot_encoder¶
Converter for scikit-learn one hot encoder.
- hummingbird.ml.operator_converters.sklearn.one_hot_encoder.convert_sklearn_one_hot_encoder(operator, device, extra_config)[source]¶
Converter for sklearn.preprocessing.OneHotEncoder
- Args:
operator: An operator wrapping a sklearn.preprocessing.OneHotEncoder 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