MLflow 更改实验 ID

Question

我无法更改 MLflow 实验的实验 ID。

目前，我运行以下代码在登录前创建一个实验：

mlflow.set_experiment(experiment_name="my_model")

with mlflow.start_run():
   #train model

这样做可以让我创建一个新实验，但实验 ID 将始终为 1。

创建的 yaml 文件如下所示：

artifact_location: file:///project/src/mlruns/1
experiment_id: '1'
lifecycle_stage: active
name: my_model

我试图查看 MLflow 文档，但我找不到更改实验 ID 的示例或函数。

如果有任何帮助或提示，我将不胜感激。

Answer 1

你应该在 start_run() 中称呼你 experiment_id:

mlflow.set_experiment("experiment name")
experiment = mlflow.get_experiment_by_name("experiment name")

with mlflow.start_run(experiment_id=experiment.experiment_id):
     # train model

注意：如果你使用set_tracking_uri()，你应该set_experiment()之后。

Answer 2

mlflow 每个实验都有 唯一 ID，如果你想要新实验 ID，你必须设置新实验名称

with mlflow.start_run(mlflow.set_experiment("new_experiment"),run_name='train') as run :
   #train model

run_name可以是任何你喜欢的，对你有意义的

如果您想更改 experiment_name="my_model" 的 exp id，请备份并删除工件和存储其映射的数据库，然后重新运行您的模块。但在删除之前确保你在 ml UI.

上看到一些其他的 exp

MLflow 更改实验 ID

MLflow change experiment id

python

machine-learning

mlflow