hummingbird.ml.operator_converters._one_hot_encoder_implementations

Base classes for one hot encoder implementations.

class hummingbird.ml.operator_converters._one_hot_encoder_implementations.OneHotEncoder(*args: Any, **kwargs: Any)[source]

Bases: PhysicalOperator, Module

Class implementing OneHotEncoder operators for ints in PyTorch.

_abc_impl = <_abc_data object>
forward(*x)[source]
class hummingbird.ml.operator_converters._one_hot_encoder_implementations.OneHotEncoderString(*args: Any, **kwargs: Any)[source]

Bases: PhysicalOperator, Module

Class implementing OneHotEncoder operators for strings in PyTorch.

Because we are dealing with tensors, strings require additional length information for processing.

_abc_impl = <_abc_data object>
forward(x)[source]