hummingbird.ml.operator_converters._mlp_implementations

Base class for multi-layer perceptrons (MLP) implementation.

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

Bases: MLPModel

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

Bases: PhysicalOperator, Module

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