python - 如何在 MLFlow 中的实验级别设置标签

Question

我可以看到 MLFlow 中的实验可以有标签（就像运行可以有标签一样）。我可以使用 设置运行标签mlflow.set_tag，但是如何为实验设置它？

标签： pythonmlflow

Answer 1

如果您查看 Python API，第一个展示如何创建的示例实际上展示了如何使用函数 ( doc ) 标记实验：mlflow.tracking packageMLflowClientclient.set_experiment_tag

from mlflow.tracking import MlflowClient

# Create an experiment with a name that is unique and case sensitive.
client = MlflowClient()
experiment_id = client.create_experiment("Social NLP Experiments")
client.set_experiment_tag(experiment_id, "nlp.framework", "Spark NLP")

您还可以使用set_model_version_tag功能为模型版本设置它，并使用set_registered_model_tag为注册模型设置它。