"""
Uses Keras's API to create a model classifier
"""
[docs]__author__ = "Elisha Yadgaran"
# Import optional dependencies
from simpleml.imports import Model
from simpleml.models.classifiers.external_models import ClassificationExternalModelMixin
from simpleml.models.classifiers.keras.base_keras_classifier import KerasClassifier
[docs]class WrappedKerasModelClassifier(Model, ClassificationExternalModelMixin):
pass
[docs]class KerasModelClassifier(KerasClassifier):
[docs] def _create_external_model(self, **kwargs):
external_model = WrappedKerasModelClassifier
return self.build_network(external_model, **kwargs)