"""
Wrapper module around `sklearn.mixture`
"""
[docs]__author__ = "Elisha Yadgaran"
from sklearn.mixture import BayesianGaussianMixture, GaussianMixture
from simpleml.models.classifiers.external_models import ClassificationExternalModelMixin
from .base_sklearn_classifier import SklearnClassifier
"""
Gaussian Mixture
"""
[docs]class WrappedSklearnBayesianGaussianMixture(
BayesianGaussianMixture, ClassificationExternalModelMixin
):
pass
[docs]class SklearnBayesianGaussianMixture(SklearnClassifier):
[docs] def _create_external_model(self, **kwargs):
return WrappedSklearnBayesianGaussianMixture(**kwargs)
[docs]class WrappedSklearnGaussianMixture(GaussianMixture, ClassificationExternalModelMixin):
pass
[docs]class SklearnGaussianMixture(SklearnClassifier):
[docs] def _create_external_model(self, **kwargs):
return WrappedSklearnGaussianMixture(**kwargs)