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

你应该叫你experiment_id在start_run()里

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，如果您想要新的exp 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上看到了其他一些实例。