python - 如何在 mlflow.start_run() 中获取当前的 run_id?

标签 python mlflow

mlflow.active_run()什么都不返回所以我不能只使用current_rui_id = mlflow.active_run().info.run_id
我必须得到 run_id 在此构造内部,以便能够继续记录另一个块内的参数、指标和工件,但对于同一模型:

with mlflow.start_run(run_name="test_ololo"):

    """ 
       fitting a model here ...
    """

    for name, val in metrics:
        mlflow.log_metric(name, np.float(val))

    # Log our parameters into mlflow
    for k, v in params.items():
        mlflow.log_param(key=k, value=v)

    pytorch.log_model(learn.model, f'model')
    mlflow.log_artifact('./outputs/fig.jpg')

我必须得到当前 run_id 在同一次运行中继续训练
with mlflow.start_run(run_id="215d3a71925a4709a9b694c45012988a"):

    """
       fit again
       log_metrics
    """

    pytorch.log_model(learn.model, f'model')
    mlflow.log_artifact('./outputs/fig2.jpg')

最佳答案

with mlflow.start_run(run_name="test_ololo") as run:

    run_id = run.info.run_id
    mlflow.log_metric("metric_1", metric_val)

关于python - 如何在 mlflow.start_run() 中获取当前的 run_id?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/59827478/

相关文章:

python - 如何从列名作为字符串的数据框中创建稀疏矩阵

docker - 如何使用 Docker 提供自定义 MLflow 模型?

python - 使用 mlflow keras-model 存储缩放器

python - 如何在运行 MLflow 的服务器上存储工件

python - 在 init 方法中将变量分配为 None 时抑制 PyCharm 警告

python - Pyinstaller:当我使用 pyinstaller 创建 .exe 文件时,使用 python 中的内置函数打开的可执行文件将在 1 秒后关闭

python - 使用复合返回语句对字典和列表的字典进行排序

python - 等效于 csv 文件的 Keras ImageDataGenerator

databricks - 在 pyspark/python 作业中访问 databricks secret

python - MLFlow 和 Hydra 一起使用会导致崩溃