Source code for simpleml.models.classifiers.sklearn.neighbors

'''
Wrapper module around `sklearn.neighbors`
'''

[docs]__author__ = 'Elisha Yadgaran'


from .base_sklearn_classifier import SklearnClassifier
from simpleml.models.classifiers.external_models import ClassificationExternalModelMixin

from sklearn.neighbors import KNeighborsClassifier


'''
K Neighbors
'''

[docs]class WrappedSklearnKNeighborsClassifier(KNeighborsClassifier, ClassificationExternalModelMixin):
    pass

[docs]class SklearnKNeighborsClassifier(SklearnClassifier):
[docs]    def _create_external_model(self, **kwargs):
        return WrappedSklearnKNeighborsClassifier(**kwargs)