hummingbird.ml.operator_converters.sklearn.mixture

hummingbird.ml.operator_converters.sklearn.mixture.convert_sklearn_BayesianGaussianMixture(operator, device, extra_config)

Converter for sklearn.mixture.BayesianGaussianMixture

Args:

operator: An operator wrapping a sklearn.mixture.BayesianGaussianMixture model device: String defining the type of device the converted operator should be run on extra_config: Extra configuration used to select the best conversion strategy

Returns:

A PyTorch model