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]¶