simpleml.models.classifiers.keras.model

Uses Keras’s API to create a model classifier

Module Contents

Classes

KerasModelClassifier

Base Keras model class. Keras objects are incrementally structured until

WrappedKerasModelClassifier

Factory class to generate library specific import wrappers

Attributes

__author__

simpleml.models.classifiers.keras.model.__author__ = Elisha Yadgaran[source]
class simpleml.models.classifiers.keras.model.KerasModelClassifier(use_training_generator=False, training_generator_params=None, use_validation_generator=False, validation_generator_params=None, use_sequence_object=False, **kwargs)[source]

Bases: simpleml.models.classifiers.keras.base_keras_classifier.KerasClassifier

Base Keras model class. Keras objects are incrementally structured until fit. Also dont have separable params. Class hijacks params to store fit params instead (enables full specification on init for reproducibility)

Pass default save method as Keras’s persistence pattern

Parameters

  • use_training_generator (Bool) – Whether to propagate use of a generator object when training – does not allow for using a generator in production – only fit_generator

  • use_validation_generator (Bool) – Whether to ALSO use a generator for validation data while training. Does nothing if use_training_generator is false

  • training_generator_params – parameters to pass to the generator method for train split - normal fit(_generator) params should be passed as params={}

  • validation_generator_params – parameters to pass to the generator method for validation split - normal fit(_generator) params should be passed as params={}

_create_external_model(self, **kwargs)[source]

Abstract method for each subclass to implement should return the desired model object

Must return external_file

Keras pattern is: external_model = SomeWrappedKerasClass(**kwargs) return self.build_network(external_model)

class simpleml.models.classifiers.keras.model.WrappedKerasModelClassifier[source]

Bases: simpleml.imports.Model, simpleml.models.classifiers.external_models.ClassificationExternalModelMixin

Factory class to generate library specific import wrappers