"""
Wrapper module around `sklearn.gaussian_process`
"""
[docs]__author__ = "Elisha Yadgaran"
from sklearn.gaussian_process import GaussianProcessClassifier
from simpleml.models.classifiers.external_models import ClassificationExternalModelMixin
from .base_sklearn_classifier import SklearnClassifier
"""
Gaussian Process Classifier
"""
[docs]class WrappedSklearnGaussianProcessClassifier(
GaussianProcessClassifier, ClassificationExternalModelMixin
):
pass
[docs]class SklearnGaussianProcessClassifier(SklearnClassifier):
[docs] def _create_external_model(self, **kwargs):
return WrappedSklearnGaussianProcessClassifier(**kwargs)