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

"""
Base module for keras classifier models
"""

[docs]__author__ = "Elisha Yadgaran"


import numpy as np

from simpleml.models.base_keras_model import KerasModel
from simpleml.models.classifiers.classification_mixin import ClassificationMixin


[docs]class KerasClassifier(KerasModel, ClassificationMixin):
[docs]    def _predict(self, X, **kwargs):
        """
        Keras returns class tuples (proba equivalent) so cast to single prediction
        """
        return np.argmax(self.external_model.predict(X), axis=1)