Source code for simpleml.models.classifiers.keras.sequential

"""
Uses Keras's API to create a sequential classifier
"""

[docs]__author__ = "Elisha Yadgaran"

# Import optional dependencies
from simpleml.imports import Sequential
from simpleml.models.classifiers.external_models import ClassificationExternalModelMixin
from simpleml.models.classifiers.keras.base_keras_classifier import KerasClassifier


[docs]class WrappedKerasSequentialClassifier(Sequential, ClassificationExternalModelMixin):
    pass


[docs]class KerasSequentialClassifier(KerasClassifier):
[docs]    def _create_external_model(self, **kwargs):
        external_model = WrappedKerasSequentialClassifier(**kwargs)
        return self.build_network(external_model, **kwargs)