hummingbird.ml.operator_converters._discretizer_implementations

Base classes for sklearn discretizers: Binarizer, KBinsDiscretizer

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

Bases: PhysicalOperator, Module

Class implementing Binarizer operators in PyTorch.

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

Bases: PhysicalOperator, Module

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