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

'''
Wrapper module around `sklearn.linear_model`
'''

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

from sklearn.linear_model import LogisticRegression, LogisticRegressionCV, Perceptron,\
    RidgeClassifier, RidgeClassifierCV, SGDClassifier


__author__ = 'Elisha Yadgaran'


'''
Logistic Regression
'''

[docs]class WrappedSklearnLogisticRegression(LogisticRegression, ClassificationExternalModelMixin):
[docs]    def get_feature_metadata(self, features, **kwargs):
        pass

[docs]class SklearnLogisticRegression(SklearnClassifier):
    def _create_external_model(self, **kwargs):
        return WrappedSklearnLogisticRegression(**kwargs)

[docs]class WrappedSklearnLogisticRegressionCV(LogisticRegressionCV, ClassificationExternalModelMixin):
[docs]    def get_feature_metadata(self, features, **kwargs):
        pass

[docs]class SklearnLogisticRegressionCV(SklearnClassifier):
    def _create_external_model(self, **kwargs):
        return WrappedSklearnLogisticRegressionCV(**kwargs)


'''
Perceptron
'''

[docs]class WrappedSklearnPerceptron(Perceptron, ClassificationExternalModelMixin):
[docs]    def get_feature_metadata(self, features, **kwargs):
        pass

[docs]class SklearnPerceptron(SklearnClassifier):
    def _create_external_model(self, **kwargs):
        return WrappedSklearnPerceptron(**kwargs)


'''
Ridge Classifier
'''

[docs]class WrappedSklearnRidgeClassifier(RidgeClassifier, ClassificationExternalModelMixin):
[docs]    def get_feature_metadata(self, features, **kwargs):
        pass

[docs]class SklearnRidgeClassifier(SklearnClassifier):
    def _create_external_model(self, **kwargs):
        return WrappedSklearnRidgeClassifier(**kwargs)

[docs]class WrappedSklearnRidgeClassifierCV(RidgeClassifierCV, ClassificationExternalModelMixin):
[docs]    def get_feature_metadata(self, features, **kwargs):
        pass

[docs]class SklearnRidgeClassifierCV(SklearnClassifier):
    def _create_external_model(self, **kwargs):
        return WrappedSklearnRidgeClassifierCV(**kwargs)


'''
SGD Classifier
'''

[docs]class WrappedSklearnSGDClassifier(SGDClassifier, ClassificationExternalModelMixin):
[docs]    def get_feature_metadata(self, features, **kwargs):
        pass

[docs]class SklearnSGDClassifier(SklearnClassifier):
    def _create_external_model(self, **kwargs):
        return WrappedSklearnSGDClassifier(**kwargs)