hummingbird.ml.operator_converters._label_encoder_implementations

Base classes for LabelEncoders

class hummingbird.ml.operator_converters._label_encoder_implementations.NumericLabelEncoder(*args: Any, **kwargs: Any)

Bases: PhysicalOperator, Module

_abc_impl = <_abc_data object>

forward(x)

class hummingbird.ml.operator_converters._label_encoder_implementations.StringLabelEncoder(*args: Any, **kwargs: Any)

Bases: PhysicalOperator, Module

LabelEncoder over string data types. When the ONNX backend is selected, this operator only works for PyTorch => 1.8.0.

_abc_impl = <_abc_data object>

forward(x)