mb.add_models(models, ...)
Adds one or more models to the model registry. The registry is updated with the new model if a an entry with that name already exists.
Parameters
models
:Dict[str, Any]
A dictionary of{ modelName: modelVariable }
. The keys of the dictionary must be strings. The values are the models to store. The values are Python variables, not paths to.pkl
files.metrics
:Optional[str, Optional[Dict[str, Any]]]
Optional, the metrics to associate with the models getting added. Like themodels
parameter, themetrics
parameter is a dictionary of{ modelName: metricsDict }
. Metrics must be JSON-serializable withstr
values for keys.serializer
:Optional["cloudpickle"]
Optional, specify a different serializer to use. Default value isNone
which uses thepickle
module. This setting applies to all models in this call. Learn mode
Returns
No value is returned. A success status message is printed if the command is successful.
Examples
Adding a model
mb.add_models({ "example_model": my_model })
Adding multiple models
mb.add_models({ "example_model": my_model, "another_model": other_model })
Adding multiple model to the "finance" directory
mb.add_models({ "finance/example_model": my_model, "finance/another_model": other_model })
Adding multiple models with metrics
mb.add_models({
"example_model": my_model,
"another_model": other_model
},
metrics={
"example_model": { "precision": 0.95 },
"another_model": { "precision": 0.80 }
})
Adding multiple models with cloudpickle
mb.add_models({ "example_model": my_model, "another_model": other_model }, serializer="cloudpickle")