Model Zoo

Model Management

ModelZoo.add_model	Add a new model together with its name to models dictionary.
ModelZoo.get_model	Return a model object.
ModelZoo.list_model_names	Return the list of model names.

Model Training

ModelZoo.train	Train a model object.
ModelZoo.train_all	Train all models.
ModelZoo.leaderboard	Show the leaderboard of all models.

Model Registry

ModelZoo.register	Register a model into MLFlow.
ModelZoo.load_registered_model	Return the list all registered models.
ModelZoo.list_registered_models	Return the list all registered models.
ModelZoo.delete_registered_model	Delete registered model.

Properties

ModelZoo.dataset	Return the dataset object.
ModelZoo.models	Return the model dictionary.