florist.api.models.models module¶ Functions and definitions for models and the Model enumeration. class Model(value)[source]¶ Bases: Enum Enumeration of supported models. MNIST = 'MNIST'¶ get_model_class()[source]¶ Return the class for this model. Return type: type[LocalDataModel] Returns: (type[LocalDataModel]) A LocalDataModel class corresponding to the model. Raises: ValueError – if the model is not supported. classmethod list()[source]¶ List all the supported models. Return type: list[str] Returns: (list[str]) a list of supported models.