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

"""
Wrapper module around `sklearn.multioutput`
"""

[docs]__author__ = "Elisha Yadgaran"


from sklearn.multioutput import ClassifierChain, MultiOutputClassifier

from simpleml.models.classifiers.external_models import ClassificationExternalModelMixin

from .base_sklearn_classifier import SklearnClassifier

"""
Classifier Chain
"""


[docs]class WrappedSklearnClassifierChain(ClassifierChain, ClassificationExternalModelMixin):
    pass


[docs]class SklearnClassifierChain(SklearnClassifier):
[docs]    def _create_external_model(self, **kwargs):
        return WrappedSklearnClassifierChain(**kwargs)


"""
Multi Output Classifier
"""


[docs]class WrappedSklearnMultiOutputClassifier(
    MultiOutputClassifier, ClassificationExternalModelMixin
):
    pass


[docs]class SklearnMultiOutputClassifier(SklearnClassifier):
[docs]    def _create_external_model(self, **kwargs):
        return WrappedSklearnMultiOutputClassifier(**kwargs)