Log and Get Models
Model comprises of model file and some metadata. Each Model can have multiple versions.
You can automatically serialize, save and version model objects as using the log_model
method.
This is an example of serializing and storing an sklearn
model. To log a model we start a run and then give our model a name
and pass in the model object and the framework
name.
import mlfoundry
import numpy as np
from sklearn.pipeline import make_pipeline
from sklearn.preprocessing import StandardScaler
from sklearn.svm import SVC
client = mlfoundry.get_client()
client.create_ml_repo("iris-demo")
run = client.create_run(ml_repo="iris-demo")
X = np.array([[-1, -1], [-2, -1], [1, 1], [2, 1]])
y = np.array([1, 1, 2, 2])
clf = make_pipeline(StandardScaler(), SVC(gamma='auto'))
clf.fit(X, y)
model_version = run.log_model(
name="iris-classifier",
model=clf,
framework="sklearn",
description="sample iris pipeline model",
metadata={"accuracy": 0.99, "f1": 0.80},
step=1, # step number, useful when using iterative algorithms like SGD
)
print(model_version.fqn)
run.end()
This will create a new model iris-demo
under the ml_repo and the first version v1
for iris-classifier
. Once created the model version is immutable.
Once created, a model version has a fqn
(fully qualified name) which can be used to retrieve the model later - E.g. model:truefoundry/user/iris-demo/iris-classifier:1
Any subsequent calls to log_model
with the same name
would create a new version of this model - v2
, v3
and so on.
The logged model can be found in the dashboard in the Models
tab under your ml_repo.

Models Tab
You can view the details of each model version from there on.

Model Version Details
Access Model
You can first get the model using the fqn
and then use the load()
function to deserialize and load the logged model.
You can also download the logged model using the fqn
and then use the download()
function. From here on you can access the files at download_info.download_dir
import mlfoundry
client = mlfoundry.get_client()
model_version = client.get_model("<fqn-of-your-model>") # e.g. "model:truefoundry/user/iris-demo/iris-classifier:1"
model = model_version.load()
download_info = model_version.download(path="path/to/your/location")
print(download_info.download_dir)
FAQs
What are the frameworks supported by the log_model
method?
log_model
method?This method supports, "sklearn", "tensorflow", "pytorch", "keras", "xgboost", "lightgbm", "fastai", "h2o", "spacy", "statsmodels", "gluon", "paddle". These options are also available as a enum - mlfoundry.ModelFramework
Updated about 2 months ago