import keras
from .models_factory import ModelsFactory
class KerasModelsFactory(ModelsFactory):
@staticmethod
def get_kwargs():
return {
'backend': keras.backend,
'layers': keras.layers,
'models': keras.models,
'utils': keras.utils,
}
Classifiers = KerasModelsFactory()