我无法更改 MLflow 实验的实验 ID。
目前,我在记录日志之前运行以下代码来创建实验:
mlflow.set_experiment(experiment_name="my_model")with mlflow.start_run(): #train model
这样做可以让我创建一个新的实验,但实验 ID 总是 1。
创建的 yaml 文件看起来像这样:
artifact_location: file:///project/src/mlruns/1experiment_id: '1'lifecycle_stage: activename: my_model
我已经查看了 MLflow 的文档,但找不到更改实验 ID 的示例或函数。
对此的任何帮助或建议我将不胜感激。
回答:
您应该在 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()
。