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()在此之后。