Accessing production-ready models#
You can access and download model files based on their stage using the
The following example illustrates how to download production-stage model files.
- There exists a model with the ID
CLS-PRE, which has at least one version with the stage set to
- There are model binaries stored in the
"model/binary"field of the model version.
Initialize a registered model and fetch its versions:
model = neptune.init_model( with_id="CLS-PRE", # (1)! ) model_versions_df = model.fetch_model_versions_table().to_pandas()
- The Neptune ID of the model
How do I find the ID?
The Neptune ID is a unique identifier for the object. It's displayed in the Details view and in the leftmost column of the table views.
The ID is stored in the system namespace. You can obtain it with
object["sys/id"].fetch(). For example:
Filter the pandas DataFrame for model versions with the stage set to "production":
Initialize each model version and download the files from the
Stop the run when done
Once you are done logging, you should stop the Neptune run. You need to do this manually when logging from a Jupyter notebook or other interactive environment:
If you're running a script, the connection is stopped automatically when the script finishes executing. In notebooks, however, the connection to Neptune is not stopped when the cell has finished executing, but rather when the entire notebook stops.