hummingbird.ml.operator_converters._mixture_implementations

Base class for Bayesian Gaussian Mixture model implementation: (BayesianGaussianMixture).

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

Bases: PhysicalOperator, Module

_abc_impl = <_abc_data object>
_estimate_log_prob(X)[source]
_estimate_log_weights()[source]
_estimate_weighted_log_prob(X)[source]
forward(X)[source]
hummingbird.ml.operator_converters._mixture_implementations._compute_log_det_cholesky(matrix_chol, covariance_type, n_features)[source]
hummingbird.ml.operator_converters._mixture_implementations._compute_precision_cholesky(covariances, covariance_type)[source]
hummingbird.ml.operator_converters._mixture_implementations._estimate_log_gaussian_prob(X, means, precisions_chol, covariance_type)[source]