ワークスペースモデルレジストリ (レガシ) を使用したモデルのライフサイクルの管理

重要

このドキュメントでは、ワークスペースモデルレジストリについて説明します。ワークスペースで Unity Catalog が有効になっている場合は、このページの手順を使用しないでください。代わりに、 Unity Catalog のモデルを参照してください。

ワークスペースモデルレジストリから Unity Catalogにアップグレードする方法のガイダンスについては、「ワークフローとモデルを Unity Catalogに移行する」を参照してください。

2024 年 4 月以降、Databricks Workspace Model Registryワークスペースのデフォルトカタログがにある新しいアカウントのワークスペースのUnity Catalog が無効になります。

この記事では、機械学習ワークフローの一部としてワークスペースモデルレジストリを使用して、ML モデルのライフサイクル全体を管理する方法について説明します。ワークスペースモデルレジストリは、Databricksが提供するMLflowモデルレジストリののホスト型バージョンです。

このWorkspace Model Registry MLflowは、MLflow 2.x と同様に 3 でも引き続きサポートされます。MLflow3 では、デフォルトレジストリ URI はdatabricks-uc で、のMLflow Model Registry Unity Catalogが使用されます。Workspace Model Registry を使用するには、 mlflow.set_registry_uri("databricks")を呼び出す必要があります。詳細については、「モデルレジストリ」を参照してください。

ワークスペースモデルレジストリには、次の機能があります。

時系列のモデルリネージ（どのMLflowのエクスペリメントとランによって、ある時点のモデルが生成されたか）。
モデルのバージョン管理。
ステージの移行 (ステージングから本番運用、アーカイブなど)。
Webhookを使用すると、レジストリイベントに基づいてアクションを自動的にトリガーできます。
モデルイベントのEメール通知。

モデルの説明を作成して表示したり、コメントを残したりすることもできます。

この記事には、ワークスペースモデルレジストリUI とワークスペースモデルレジストリAPI の両方の手順が含まれています。

Workspace Model Registryの概念の概要については、 DatabricksのMLflow参照してください。

モデルを作成または登録する

UI を使用してモデルを作成または登録するか、APIを使用してモデルを登録することができます。

UI を使用してモデルを作成または登録する

ワークスペースモデルレジストリにモデルを登録するには、2つの方法があります。 MLflowに記録された既存のモデルを登録するか、新しい空のモデルを作成して登録し、以前に記録済みモデルを割り当てることができます。

既存の記録済みモデルをノートブックから登録する

ワークスペースで、登録するモデルを含む MLflow 実行を特定します。
1. ノートブックの右側のサイドバーにある エクスペリメント アイコンをクリックします。
2. エクスペリメント実行サイドバーで、実行日の横にあるアイコンをクリックします。 [MLflow の実行] ページが表示されます。このページには、パラメーター、メトリクス、タグ、アーティファクトのリストなど、実行の詳細が表示されます。
「アーティファクト」セクションで、 xxx-model という名前のディレクトリをクリックします。
右端の モデルを登録する ボタンをクリックします。
ダイアログで、[ モデル ]ボックスをクリックし、次のいずれかの操作を行います。
- ドロップダウンメニューから[ 新しいモデルを作成 ]を選択します。 「モデル名 」フィールドが表示されます。モデル名を入力します (例: scikit-learn-power-forecasting)。
- ドロップダウンメニューから既存のモデルを選択します。
登録をクリックします。
- [ 新しいモデルの作成 ] を選択した場合は、scikit-learn-power-forecasting という名前のモデルが登録され、ワークスペースモデルレジストリによって管理される安全な場所にモデルがコピーされ、モデルの新しいバージョンが作成されます。
- 既存のモデルを選択した場合は、選択したモデルの新しいバージョンが登録されます。
しばらくすると、[ モデルを登録する ] ボタンが新しい登録済みモデルバージョンへのリンクに変わります。
リンクをクリックして、ワークスペースモデルレジストリ UIで新しいモデルバージョンを開きます。ワークスペースモデルレジストリでモデルを見つけるには、サイドバーの モデル をクリックします。

新しい登録済みモデルを作成し、それに記録済みモデルを割り当てる

[登録済みモデル] ページの [モデルの作成] ボタンを使用して、新しい空のモデルを作成し、それに記録済みモデルを割り当てることができます。以下の手順に従います。

登録済みモデルのページで、[ モデルの作成 ] をクリックします。モデルの名前を入力し、[ 作成 ]をクリックします。
ノートブックから既存の記録済みモデルを登録するの手順 1 から 3 に従います。
[登録する Model] ダイアログで、ステップ 1 で作成したモデルの名前を選択し、[ 登録する ] をクリックします。これにより、作成した名前のモデルが登録され、ワークスペースモデルレジストリによって管理される安全な場所にモデルがコピーされ、モデルバージョン (Version 1) が作成されます。

しばらくすると、 MLflow 実行 UI の [登録する Model] ボタンが、新しく登録されたモデルバージョンへのリンクに置き換わります。これで、 エクスペリメントラン ページの「モデルを登録する」ダイアログの モデル ドロップダウンリストからモデルを選択できるようになりました。また、Create ModelVersion などの API コマンドでモデルの名前を指定して、モデルの新しいバージョンを登録することもできます。

API を使用してモデルを登録する

ワークスペースモデルレジストリにモデルを登録するには、プログラムによる方法が3つあります。すべての方法で、ワークスペースモデルレジストリによって管理される安全な場所にモデルがコピーされます。

MLflowエクスペリメント中にモデルをログに記録し、指定した名前で登録するには、mlflow.<model-flavor>.log_model(...) メソッドを使用します。名前が登録されたモデルが存在しない場合、メソッドは新しいモデルを登録し、バージョン 1 を作成し、 ModelVersion MLflow オブジェクトを返します。その名前の登録済みモデルがすでに存在する場合、メソッドは新しいモデルバージョンを作成し、バージョンオブジェクトを返します。
Python
```
with mlflow.start_run(run_name=<run-name>) as run:
  ...
  mlflow.<model-flavor>.log_model(<model-flavor>=<model>,
    artifact_path="<model-path>",
    registered_model_name="<model-name>"
  )
```
指定した名前のモデルを登録するには、すべてのエクスペリメントの実行が完了し、レジストリに追加するのに最も適したモデルを決定した後、 mlflow.register_model() 方法を使用します。この方法では、 mlruns:URI 引数の実行 ID が必要です。名前が登録されたモデルが存在しない場合、メソッドは新しいモデルを登録し、バージョン 1 を作成し、 ModelVersion MLflow オブジェクトを返します。その名前の登録済みモデルがすでに存在する場合、メソッドは新しいモデルバージョンを作成し、バージョンオブジェクトを返します。
Python
```
result=mlflow.register_model("runs:<model-path>", "<model-name>")
```
指定した名前で新しい登録済みモデルを作成するには、MLflow クライアント API create_registered_model() メソッドを使用します。モデル名が存在する場合、このメソッドは MLflowExceptionをスローします。
Python
```
client = MlflowClient()
result = client.create_registered_model("<model-name>")
```

Databricks Terraform プロバイダーと databricks_mlflow_model にモデルを登録することもできます。

クォータ制限

2024 年 5 月以降、すべての Databricks ワークスペースに対して、ワークスペースモデルレジストリワークスペースごとの登録済みモデルとモデルバージョンの合計数にクォータ制限が課されます。リソースの制限を参照してください。レジストリクォータを超えた場合、Databricks では、登録されたモデルと不要になったモデルバージョンを削除することをお勧めします。 Databricks では、モデルの登録と保持の戦略を調整して、制限内に収まるようにすることもお勧めします。ワークスペースの制限を増やす必要がある場合は、Databricks アカウントチームにお問い合わせください。

次のノートブックは、モデルレジストリエンティティをインベントリおよび削除する方法を示しています。

ワークスペースモデルレジストリエンティティを一覧するノートブック

ノートブックを新しいタブで開く

UIでモデルを表示する

登録済みモデルページ

サイドバーの モデル をクリックすると、「登録済みモデル」ページが表示されます。このページには、レジストリ内のすべてのモデルが表示されます。

このページから新しいモデルを作成できます。

また、このページから、ワークスペース管理者はワークスペースモデルレジストリ内のすべてのモデルに権限を設定できます。

登録済みモデル

登録モデルページ

モデルの登録済みモデルページを表示するには、登録済みモデルページでモデル名をクリックします。登録済みモデルページには、選択したモデルに関する情報と、モデルの各バージョンに関する情報を含むテーブルが表示されます。このページでは、次のこともできます。

推論にモデルを使用するノートブックを自動的に生成します。
Eメール通知を設定します。
モデルのバージョンを比較します。
モデルの権限を設定します。
モデルを削除します。

登録されたモデル

モデルバージョンページ

モデルバージョンページを表示するには、次のいずれかの操作を行います。

登録済みモデルページの 最新バージョン 列でバージョン名をクリックします。
登録済みモデルページの バージョン 列でバージョン名をクリックします。

このページには、登録済みモデルの特定のバージョンに関する情報が表示され、ソース実行 (モデルを作成するために実行されたノートブックのバージョン) へのリンクも表示されます。このページでは、次のこともできます。

推論にモデルを使用するノートブックを自動的に生成します。
モデルを削除します。

モデルバージョン

モデルへのアクセスの制御

モデルに対する権限を設定するには、少なくともCAN MANAGE権限が必要です。モデルのアクセス許可レベルに関する情報については、「MLflowモデル ACL」を参照してください。モデルバージョンは、親モデルから権限を継承します。モデルバージョンの権限は設定できません。

サイドバーで、「 モデル 」をクリックします。
モデル名を選択します。
[権限] をクリックします。[権限設定] ダイアログが開きます
ダイアログで、[ ユーザー、グループ、またはサービスプリンシパルの選択 ] ドロップダウンを選択し、ユーザー、グループ、またはサービスプリンシパルを選択します。
権限ドロップダウンから権限を選択します。
追加をクリックし、保存をクリックします。

ワークスペース管理者とレジストリ全体のレベルで CAN MANAGE 権限を持つユーザーは、[モデル] ページの アクセス許可 をクリックして、ワークスペース内のすべてのモデルにアクセス許可レベルを設定できます。

モデルステージの移行

モデルバージョンには、なし、 ステージング 、 本番運用 、 アーカイブ のいずれかのステージがあります。 ステージング ステージはモデルのテストと検証を目的としており、 本番運用 ステージは、テストまたはレビュープロセスを完了し、ライブスコアリングのためにアプリケーションにデプロイされたモデルバージョン用です。アーカイブ済みモデルバージョンは非アクティブと見なされ、その時点で削除を検討できます。モデルの異なるバージョンは、異なる段階にある場合があります。

適切な権限を持つユーザーは、ステージ間でモデルバージョンを移行できます。モデルバージョンを特定のステージに移行する権限がある場合は、直接移行を行うことができます。権限がない場合は、ステージの移行をリクエストでき、モデルバージョンの移行権限を持つユーザーはリクエストを承認、拒否、またはキャンセルできます。

モデルステージは、UIまたは APIを使用して移行できます。

UIを使用したモデルステージのトランジション

次の手順に従って、モデルのステージを移行します。

使用可能なモデル・ステージと使用可能なオプションのリストを表示するには、モデル・バージョン・ページで、「 ステージ:」 の横にあるドロップダウンをクリックし、別のステージへの遷移をリクエストまたは選択します。
オプションのコメントを入力し、 OK をクリックします。

モデル版を本番運用ステージに移行する

テストと検証を行った後、本番運用ステージへの移行または移行のリクエストを行うことができます。

ワークスペースモデルレジストリでは、各ステージで登録済みモデルの複数のバージョンを使用できます。本番運用に1つのバージョンのみを含める場合は、「 既存の本番運用モデルのバージョンをアーカイブ済みに移行する 」にチェックを入れることで、現在本番運用中のモデルのすべてのバージョンをアーカイブ済みに移行できます。

モデルバージョンステージの移行要求を承認、拒否、またはキャンセルする

ステージの移行権限を持たないユーザーは、ステージの移行をリクエストできます。要求は、モデルバージョンページの 保留中の要求 セクションに表示されます。

本番運用への移行

ステージの移行リクエストを承認、却下、またはキャンセルするには、[ 承認 ]、[ 却下 ]、または [ キャンセル ] リンクをクリックします。

トランジションリクエストの作成者は、リクエストをキャンセルすることもできます。

モデルバージョンのアクティビティの表示

リクエストされたトランジション、承認されたトランジション、保留中のトランジション、モデルバージョンに適用されたすべてのトランジションを表示するには、[アクティビティ]セクションに移動します。このアクティビティの記録は、監査または検査のためのモデルのライフサイクルのリネージを提供します。

API を使用したモデルステージの移行

適切な権限を持つユーザーは、モデルバージョンを新しいステージに移行できます。

モデルバージョンステージを新しいステージに更新するには、MLflow クライアント API transition_model_version_stage() メソッドを使用します。

Python
  client = MlflowClient()
  client.transition_model_version_stage(
    name="<model-name>",
    version=<model-version>,
    stage="<stage>",
    description="<description>"
  )

<stage>に指定できる値は、"Staging"|"staging"、"Archived"|"archived"、"Production"|"production"、"None"|"none"です。

推論にモデルを使用する

備考

プレビュー

この機能はパブリックプレビュー段階です。

モデルをワークスペースモデルレジストリに登録した後、バッチ推論またはストリーミング推論にモデルを使用するノートブックを自動的に生成できます。

登録済みのモデルページまたはモデルバージョンページの右上隅にある[]をクリックします。[モデル推論の構成] ダイアログが表示され、バッチ推論、ストリーミング推論、またはリアルタイム推論を構成できます。

重要

株式会社Anacondaは、anaconda.org チャンネルの利用規約を更新しました。新しいサービス条件に基づき、Anacondaのパッケージングと配布に依存する場合は、商用ライセンスが必要になる場合があります。詳細については、 Anaconda Commercial Edition の FAQ を参照してください。 Anacondaチャンネルの使用は、その利用規約に準拠します。

MLflowv1.18 より前 (Databricks Runtime 8.3ML 以前) で記録されたモデルは、デフォルトによって condadefaults チャンネル (https://repo.anaconda.com/pkgs/) で記録されていました依存関係として。このライセンス変更に伴い、 Databricks は MLflow v1.18 以降を使用してログインしたモデルでの defaults チャンネルの使用を停止しました。ログに記録されたデフォルトチャンネルは、コミュニティが管理する https://conda-forge.org/ を指す conda-forgeになりました。

MLflow v1.18より前のモデルをログに記録し、そのモデルのconda環境からdefaultsチャンネルを除外しなかった場合、そのモデルは意図していないdefaultsチャンネルに依存している可能性があります。モデルにこの依存関係があるかどうかを手動で確認するには、記録済みモデルにパッケージ化されているconda.yaml ファイル内のchannel値を調べます。たとえば、defaults チャンネルの依存関係を持つモデルのconda.yamlは、次のようになります。

YAML
channels:
- defaults
dependencies:
- python=3.8.8
- pip
- pip:
    - mlflow
    - scikit-learn==0.23.2
    - cloudpickle==1.6.0
      name: mlflow-env

Databricksは、Anacondaとの関係の下で、モデルと対話するためのAnacondaリポジトリの使用が許可されているかどうかを判断できないため、Databricksは顧客に変更を加えることを強制していません。Databricks の使用による Anaconda.com リポジトリの使用が Anaconda の条件で許可されている場合は、何もする必要はありません。

モデルの環境で使用するチャンネルを変更したい場合は、新しい conda.yamlでモデルをワークスペースモデルレジストリに再登録するこれを行うには、log_model()の conda_env パラメーターでチャンネルを指定します。

log_model() APIの詳細については、使用しているモデルフレーバーのMLflowドキュメンテーション(log_model for a scikit-learnなど)を参照してください。

conda.yamlファイルの詳細については、 MLflow のドキュメントを参照してください。

モデル推論ダイアログの構成

バッチ推論を構成する

これらの手順に従ってバッチ推論ノートブックを作成すると、ノートブックは、モデルの名前が付いたフォルダー内の Batch-Inference フォルダーの下のユーザーフォルダーに保存されます。必要に応じてノートブックを編集できます。

バッチ推論 タブをクリックします。
モデルバージョン ドロップダウンから、使用するモデルバージョンを選択します。ドロップダウンの最初の 2 つの項目は、モデルの現在の本番運用バージョンとステージングバージョン (存在する場合) です。これらのオプションのいずれかを選択すると、ノートブックは実行時点の本番運用バージョンまたはステージングバージョンを自動的に使用します。モデルの開発を続行するときに、ノートブックを更新する必要はありません。
[入力テーブル ] の横にある [参照 ] ボタンをクリックします。 [入力データの選択 ] ダイアログが表示されます。必要に応じて、 コンピュート ドロップダウンでクラスターを変更できます。

注記

Unity Catalog 対応ワークスペースの場合、 入力データの選択 ダイアログでは、 <catalog-name>.<database-name>.<table-name>の 3 つのレベルから選択できます。

モデルの入力データを含むテーブルを選択し、選択をクリックします。生成されたノートブックは、このデータを自動的にインポートしてモデルに送信します。生成されたノートブックは、モデルに入力する前にデータの変換が必要な場合に編集できます。
予測は、ディレクトリの dbfs:/FileStore/batch-inferenceのフォルダに保存されます。デフォルトでは、予測はモデルと同じ名前のフォルダに保存されます。生成されたノートブックを実行するたびに、名前にタイムスタンプが付加された新しいファイルがこのディレクトリに書き込まれます。また、タイムスタンプを含めず、ノートブックの後続の実行でファイルを上書きすることもできます。手順は、生成されたノートブックに記載されています。

予測を保存するフォルダを変更するには、 出力テーブルの場所フィールドに新しいフォルダ名を入力するか 、フォルダアイコンをクリックしてディレクトリをブラウズし、別のフォルダを選択します。

予測を Unity Catalog内の場所に保存するには、ノートブックを編集する必要があります。 Unity Catalog のデータを使用する機械学習モデルをトレーニングし、結果を Unity Catalog に書き戻す方法を示すノートブックの例については、「機械学習のチュートリアル」を参照してください。

Lakeflow Spark宣言型パイプラインを使用してストリーミング推論を構成する

これらの手順に従ってストリーミング推論ノートブックを作成すると、ノートブックは、モデルの名前を持つフォルダー内の DLT-Inference フォルダーの下のユーザーフォルダーに保存されます。必要に応じてノートブックを編集できます。

[ストリーミング ( Lakeflow Spark宣言型パイプライン)] タブをクリックします。
モデルバージョン ドロップダウンから、使用するモデルバージョンを選択します。ドロップダウンの最初の 2 つの項目は、モデルの現在の本番運用バージョンとステージングバージョン (存在する場合) です。これらのオプションのいずれかを選択すると、ノートブックは実行時点の本番運用バージョンまたはステージングバージョンを自動的に使用します。モデルの開発を続行するときに、ノートブックを更新する必要はありません。
[入力テーブル ] の横にある [参照 ] ボタンをクリックします。 [入力データの選択 ] ダイアログが表示されます。必要に応じて、 コンピュート ドロップダウンでクラスターを変更できます。
モデルの入力データを含むテーブルを選択し、選択をクリックします。生成されたノートブックは、入力テーブルをソースとして使用するデータ変換を作成し、MLflow PySpark 推論 UDF を統合してモデル予測を実行します。モデルの適用前または適用後にデータに追加の変換が必要な場合は、生成されたノートブックを編集できます。
出力のLakeflow Spark宣言型パイプライン名を指定します。ノートブックは、指定された名前でライブテーブルを作成し、それを使用してモデルの予測を保存します。生成されたノートブックを変更して、必要に応じてターゲットデータセットをカスタマイズできます。たとえば、ストリーミングライブテーブルを出力として定義したり、スキーマ情報やデータ品質の制約を追加したりできます。
その後、このノートブックを使用して新しいパイプラインを作成するか、追加のノートブックライブラリとして既存のパイプラインに追加できます。

フィードバックの提供

この機能はプレビュー段階であり、フィードバックをお待ちしております。フィードバックを提供するには、[モデル推論の構成] ダイアログで [ Provide Feedback ] をクリックします。

モデルバージョンの比較

ワークスペースモデルレジストリでモデルのバージョンを比較できます。

登録モデルページで、モデルバージョンの左側にあるチェックボックスをクリックして、2つ以上のモデルバージョンを選択します。
比較をクリックします。
[ <N> バージョンの比較] 画面が表示され、選択したモデルバージョンのパラメーター、スキーマ、およびメトリクスを比較するテーブルが表示されます。画面の下部で、プロットのタイプ(散布図、等高線座標、または平行座標)と、プロットするパラメーターまたはメトリクスを選択できます。

通知設定を制御する

登録されたモデルおよび指定したモデル・バージョンでのアクティビティーについて、Eメールで通知するようにワークスペースモデルレジストリを構成できます。

登録モデルページの「 Notify me about 」メニューには、次の3つのオプションが表示されます。

すべての新しいアクティビティ : このモデルのすべてのモデルバージョンのすべてのアクティビティに関するEメール通知を送信します。登録済みのモデルを作成した場合は、この設定がデフォルトです。
フォローしているバージョンでのアクティビティ : フォローしているモデルバージョンについてのみEメール通知を送信します。この選択により、フォローしているすべてのモデルバージョンの通知を受け取ります。特定のモデルバージョンの通知をオフにすることはできません。
通知のミュート :この登録済みモデルのアクティビティに関するEメール通知を送信しません。

次のイベントがEメール通知をトリガーします。

新モデルバージョンの作成
ステージ移行のリクエスト
ステージ移行
新しいコメント

次のいずれかの操作を行うと、モデル通知が自動的にサブスクライブされます。

そのモデルバージョンについてコメントする
モデルバージョンのステージの移行
モデルのステージの遷移要求を行う

モデル・バージョンをフォローしているかどうかを確認するには、モデル・バージョン・ページの「ステータスのフォロー」フィールド、または登録済みモデル・ページのモデル・バージョンの表を確認します。

すべてのEメール通知をオフにする

Eメール通知は、[ユーザー設定] メニューの [ ワークスペースモデルレジストリ設定] タブでオフにできます。

Databricks ワークスペースの右上隅にあるユーザー名をクリックし、ドロップダウンメニューから [ 設定 ] を選択します。
設定サイドバーで、通知を選択します。
モデルレジストリのEメール通知 をオフにします。

アカウント管理者は、管理者設定ページで組織全体のEメール通知をオフにすることができます。

Eメール送信の最大数

ワークスペースモデルレジストリ、各アクティビティごとに各ユーザーに送信されるEメールの数を制限します。例えば、登録したモデルに対して作成された新しいモデルバージョンについて、1日に20通のEメールを受け取った場合、ワークスペースモデルレジストリは1日の制限に達したことを知らせるEメールを送信し、そのイベントに関する追加のEメールは翌日まで送信されません。

Eメールの上限を引き上げるには、 Databricks アカウントチームにお問い合わせください。

Webhook

備考

プレビュー

この機能はパブリックプレビュー段階です。

Webhook を使用すると、ワークスペースモデルレジストリイベントをリッスンして、インテグレーションでアクションを自動的にトリガーできます。 Webhook を使用して、機械学習パイプラインを自動化し、既存の CI/CD ツールやワークフローと統合できます。たとえば、新しいモデルバージョンが作成されたときにCIビルドをトリガーしたり、本番運用へのモデル移行がリクエストされるたびにSlackを通じてチームメンバーに通知したりできます。

モデルまたはモデルバージョンに注釈を付けてください

アノテーションによって、モデルまたはモデルのバージョンに関する情報を提供できます。たとえば、問題の概要や、使用された方法論およびアルゴリズムに関する情報を含めることができます。

UIを使用したモデルまたはモデルバージョンの注釈付け

Databricks UI には、モデルとモデルバージョンに注釈を付ける方法がいくつか用意されています。説明やコメントを使用してテキスト情報を追加したり、検索可能なキー値タグを追加したりできます。説明とタグは、モデルとモデルバージョンで使用できます。コメントはモデルバージョンでのみ使用できます。

説明は、モデルに関する情報を提供することを目的としています。
コメントは、モデルバージョンのアクティビティに関する継続的な議論を維持する方法を提供します。
タグを使用すると、モデルのメタデータをカスタマイズして、特定のモデルを見つけやすくすることができます。

モデルまたはモデルバージョンの説明を追加または更新してください

登録済みモデルまたはモデルバージョンページで、説明の横にある編集をクリックします。編集ウィンドウが表示されます。
編集ウィンドウで説明を入力または編集します。
保存をクリックして変更を保存するか、 キャンセル をクリックしてウィンドウを閉じます。

モデルバージョンの説明を入力した場合、その説明は登録済みモデルページのテーブルの「説明」列に表示されます。列には、最大 32 文字または 1 行のテキストのいずれか短い方が表示されます。

モデルバージョンのコメントを追加する

モデルバージョンページを下にスクロールし、 アクティビティ の横にある下矢印をクリックします。
編集ウィンドウにコメントを入力し、 コメントの追加 をクリックします。

モデルまたはモデルバージョンのタグを追加する

登録済みモデルまたはモデルバージョンページで、まだ開いていない場合はをクリックします。タグテーブルが表示されます。
「名前 」フィールドと「値」フィールドをクリックし、タグのキーと値を入力します。
[ 追加 ] をクリックします。

モデルまたはモデルバージョンのタグの編集または削除

既存のタグを編集または削除するには、[ アクション ] 列のアイコンを使用します。

タグアクション

API を使用してモデルバージョンに注釈を付ける

モデルバージョンの説明を更新するには、MLflowクライアントAPIのupdate_model_version()メソッドを使用します。

Python
client = MlflowClient()
client.update_model_version(
  name="<model-name>",
  version=<model-version>,
  description="<description>"
)

登録済みモデルまたはモデルバージョンのタグを設定または更新するには、MLflow クライアント API set_registered_model_tag()) または set_model_version_tag() メソッドを使用します。

Python
client = MlflowClient()
client.set_registered_model_tag()(
  name="<model-name>",
  key="<key-value>",
  tag="<tag-value>"
)

Python
client = MlflowClient()
client.set_model_version_tag()(
  name="<model-name>",
  version=<model-version>,
  key="<key-value>",
  tag="<tag-value>"
)

モデルの名前を変更する (API のみ)

登録されたモデルの名前を変更するには、MLflowクライアントAPIのrename_registered_model()メソッドを使用します。

Python
client=MlflowClient()
client.rename_registered_model("<model-name>", "<new-model-name>")

注記

登録済みモデルの名前を変更できるのは、バージョンがない場合、またはすべてのバージョンが [なし] ステージまたは [アーカイブ済み] ステージにある場合のみです。

モデルを検索する

ワークスペースモデルレジストリでモデルは、UI または API を使用して検索できます。

注記

モデルを検索すると、少なくとも CAN READ パーミッションを持つモデルのみが返されます。

UI を使用したモデルの検索

登録済みのモデルを表示するには、サイドバーの「 モデル」をクリックします。

特定のモデルを検索するには、検索ボックスにテキストを入力します。モデルの名前または名前の任意の部分を入力できます。

登録機種検索

タグで検索することもできます。タグは tags.<key>=<value>の形式で入力します。複数のタグを検索するには、 AND 演算子を使用します。

MLflow 検索構文を使用して、モデル名とタグの両方を検索できます。例えば：

API を使用したモデルの検索

MLflow search_registered_models()メソッドを使用して、 Workspace Model Registry内の登録済みモデルを検索できます。

モデルにタグを設定している場合は、search_registered_models() を使用してそれらのタグで検索することもできます。

Python
import mlflow
from pprint import pprint

print(f"Find registered models with a specific tag value")
for m in mlflow.search_registered_models(f"tags.`<key-value>`='<tag-value>'"):
  pprint(dict(m), indent=4)

MLflow search_model_versions()メソッドを使用して、特定のモデル名を検索し、そのバージョンの詳細を一覧表示することもできます。

Python
import mlflow
from pprint import pprint

[pprint(mv) for mv in mlflow.search_model_versions("name='<model-name>'")]

出力は以下のようになります。

Console
{   'creation_timestamp': 1582671933246,
    'current_stage': 'Production',
    'description': 'A random forest model containing 100 decision trees '
                   'trained in scikit-learn',
    'last_updated_timestamp': 1582671960712,
    'name': 'sk-learn-random-forest-reg-model',
    'run_id': 'ae2cc01346de45f79a44a320aab1797b',
    'source': './mlruns/0/ae2cc01346de45f79a44a320aab1797b/artifacts/sklearn-model',
    'status': 'READY',
    'status_message': None,
    'user_id': None,
    'version': 1 }

{   'creation_timestamp': 1582671960628,
    'current_stage': 'None',
    'description': None,
    'last_updated_timestamp': 1582671960628,
    'name': 'sk-learn-random-forest-reg-model',
    'run_id': 'd994f18d09c64c148e62a785052e6723',
    'source': './mlruns/0/d994f18d09c64c148e62a785052e6723/artifacts/sklearn-model',
    'status': 'READY',
    'status_message': None,
    'user_id': None,
    'version': 2 }

モデルまたはモデルバージョンを削除する

モデルは、UI または API を使用して削除できます。

UI を使用したモデルのバージョンまたはモデルの削除

警告

この操作は元に戻せません。モデル・バージョンは、レジストリから削除するのではなく、アーカイブ・ステージに移行できます。モデルを削除すると、ワークスペースモデルレジストリによって保存されたすべてのモデルアーティファクトと、登録されたモデルに関連付けられているすべてのメタデータが削除されます。

注記

モデルとモデル・バージョンは、「なし」ステージまたは「アーカイブ済」ステージでのみ削除できます。登録済みモデルのバージョンがステージングステージまたは本番運用ステージにある場合は、モデルを削除する前に、それらをなしステージまたはアーカイブ済みステージに移行する必要があります。

モデルバージョンを削除するには:

サイドバーの「 モデル 」をクリックします。
モデル名をクリックします。
モデルバージョンをクリックします。
画面の右上隅にあるケバブメニューをクリックし、ドロップダウンメニューから[ 削除 ]を選択します。

モデルを削除するには:

サイドバーの「 モデル 」をクリックします。
モデル名をクリックします。
画面の右上隅にあるケバブメニューをクリックし、ドロップダウンメニューから[ 削除 ]を選択します。

API を使用してモデルのバージョンまたはモデルを削除する

警告

注記

モデルバージョンを削除します

モデルバージョンを削除するには、MLflowクライアントAPIのdelete_model_version()メソッドを使用します。

Python
# Delete versions 1,2, and 3 of the model
client = MlflowClient()
versions=[1, 2, 3]
for version in versions:
  client.delete_model_version(name="<model-name>", version=version)

モデルの削除

モデルを削除するには、MLflowクライアントAPIのdelete_registered_model()メソッドを使用します。

Python
client = MlflowClient()
client.delete_registered_model(name="<model-name>")

ワークスペース間で MLflow オブジェクトをコピーする

MLflowDatabricksワークスペースとの間でオブジェクトをインポートまたはエクスポートするには、コミュニティ主導のオープンソースプロジェクトMLflow Export-Import MLflowを使用して、ワークスペース間でエクスペリメント、モデル、および実行移行できます。

これらのツールを使用すると、次のことができます。

同じまたは別のトラッキングサーバー内の他のデータサイエンティストと共有し、共同作業します。たとえば、他のユーザーのエクスペリメントをワークスペースに複製できます。
1 つのワークスペースから別のワークスペース (開発ワークスペースから本番運用ワークスペースなど) にモデルをコピーします。
エクスペリメント MLflow をコピーして、ローカルの追跡サーバーから Databricks ワークスペースに実行します。
ミッションクリティカルなエクスペリメントとモデルを別の Databricks ワークスペースにバックアップします。

例

この例では、ワークスペースモデルレジストリを使用して機械学習アプリケーションを構築する方法を示します。

ワークスペースモデルレジストリの例

モデルを作成または登録する​

UI を使用してモデルを作成または登録する​

既存の記録済みモデルをノートブックから登録する​

新しい登録済みモデルを作成し、それに記録済みモデルを割り当てる​

API を使用してモデルを登録する​

クォータ制限​

ワークスペース モデルレジストリ エンティティを一覧するノートブック

UIでモデルを表示する​

登録済みモデルページ​

登録モデルページ​

モデルバージョンページ​

モデルへのアクセスの制御​

モデルステージの移行​

UIを使用したモデルステージのトランジション​

モデル版を本番運用ステージに移行する​

モデル バージョン ステージの移行要求を承認、拒否、またはキャンセルする​

モデルバージョンのアクティビティの表示​

API を使用したモデルステージの移行​

推論にモデルを使用する​

バッチ推論を構成する​

Lakeflow Spark宣言型パイプラインを使用してストリーミング推論を構成する​

フィードバックの提供​

モデルバージョンの比較​

通知設定を制御する​

すべてのEメール通知をオフにする​

Eメール送信の最大数​

Webhook​

モデルまたはモデルバージョンに注釈を付けてください​

UIを使用したモデルまたはモデルバージョンの注釈付け​

モデルまたはモデルバージョンの説明を追加または更新してください​

モデル バージョンのコメントを追加する​

モデルまたはモデル バージョンのタグを追加する​

モデルまたはモデル バージョンのタグの編集または削除​

API を使用してモデル バージョンに注釈を付ける​

モデルの名前を変更する (API のみ)​

モデルを検索する​

UI を使用したモデルの検索​

API を使用したモデルの検索​

モデルまたはモデルバージョンを削除する​

UI を使用したモデルのバージョンまたはモデルの削除​

API を使用してモデルのバージョンまたはモデルを削除する​

モデルバージョンを削除します​

モデルの削除​

ワークスペース間で MLflow オブジェクトをコピーする​

例​