嗨@AnnamalaiVR,
謝謝你發帖磚社區的問題。
在Python代碼中,導入MLflow圖書館和創建一個客戶端對象來訪問您的模型注冊表。
% python
進口
mlflow
#設置磚跟蹤的URI
databricks_host =
“* * * * * * * * * * * * * * * * * * * * *”
databricks_token =
“* * * * * * * * * *”
databricks_org_id =
“* * * * * *”
tracking_uri =
磚:/ / f”
{
databricks_host
}
org_id = ?
{
databricks_org_id
}
”
mlflow.set_tracking_uri
(
tracking_uri
)
#配置MLflow客戶機
客戶= mlflow.tracking.MlflowClient
()
現在你可以查詢注冊表使用模型
客戶端
對象。這裏有一個例子來獲取注冊模型版本對於一個給定的模型名稱:
% python
model_name =
“my_model”
model_versions = client.search_model_versions
(
f“name = '
{
model_name
}
’”
)
為
model_version
在
model_versions
:
#獲取運行ID和度量模型版本
run_id = model_version.run_id
指標= client.get_run
(
run_id
)
.data.metrics
#添加指標字典
accuracy_score =指標
(
“準確性分數”
]
nw_dict
(
run_id
]
= accuracy_score
確保model_name變量替換的名稱注冊模型。
