hummingbird.ml.operator_converters._nb_implementations

Base classes for Naive Bayes model implementation: (BernoulliNB, GaussianNB).

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

Bases: PhysicalOperator, Module

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

Bases: PhysicalOperator, Module

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