hummingbird.ml.operator_converters._imputer_implementations

Base classes for Imputers

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

Bases: PhysicalOperator, Module

Class implementing Imputer operators in MissingIndicator.

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

Bases: PhysicalOperator, Module

Class implementing SimpleImputer operators in PyTorch.

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