Creating a model version#
Once you have registered a model, you can create and store versions of it. This lets you track the stage, binaries, and relevant metadata of each model version separately.
Before you start
You need to have a model registered in a Neptune project. You can find your models under the Models tab.
For instructions, see Registering a model.
To create and store a model version:
-
Import Neptune and initialize a
ModelVersion
object based on the ID of an existing model:- Neptune ID of the model, which comes from the project key and model key put together. You can find it in the leftmost column of the models table.
The first model version will have the ID
CLS-PRE-1
. -
(optional) Log some model parameters and other metadata:
-
(optional) Upload the trained model weights as a binary file:
-
To stop the connection to Neptune and sync all data, call the
stop()
method and execute the script or cell:
To view the created model version in Neptune, click the link in the console output.