Create a registered model
Registered model を作成して、モデリングタスクの候補モデルを保持します。Model Registry 内で対話型で登録モデルを作成するか、Python SDK を使ってプログラム上で作成できます。
プログラムで登録モデルを作成する
W&B Python SDK を使ってプログラムでモデルを登録します。登録モデルが存在しない場合、W&B は自動的に登録モデルを作成します。
<>
で囲まれている他の値を自分の値に置き換えてください:
import wandb
run = wandb.init(entity="<entity>", project="<project>")
run.link_model(path="<path-to-model>", registered_model_name="<registered-model-name>")
run.finish()
registered_model_name
に指定した名前が Model Registry App に表示されます。
対話型で登録モデルを作成する
Model Registry App 内で対話型で登録モデルを作成します。
- https://wandb.ai/registry/model の Model Registry App に移動します。
- Model Registry ページの右上にある New registered model ボタンをクリックします。
- 表示されたパネルから、登録モデルが所属する Entity を Owning Entity ドロップダウンから選択します。
- Name フィールドにモデルの名前を入力します。
- Type ドロップダウンから、登録モデルにリンクするアーティファクトのタイプを選択します。
- (オプション)Description フィールドにモデルに関する説明を追加します。
- (オプション)Tags フィールドに1つ以上のタグを追加します。
- Register model をクリックします。
ヒント
モデルをモデルレジストリに手動でリンクするのは、一度きりのモデルに便利です。しかし、しばしばモデルバージョンをプログラムでモデルレジストリにリンクすることが有用です。
例えば、毎晩実行されるジョブがあるとします。毎晩作成されるモデルを手動でリンクするのは面倒です。その代わりに、モデルを評価し、パフォーマンスが向上した場合に W&B Python SDK でそのモデルをモデルレジストリにリンクするスクリプトを作成できます。