From 384cf56b89e1ed40112bb3b8e599d2e92e579bb7 Mon Sep 17 00:00:00 2001 From: jj Date: Mon, 5 Feb 2024 11:55:08 +0100 Subject: [PATCH] two apis quick hack --- model_search_upload.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/model_search_upload.py b/model_search_upload.py index 35727ad..51467ac 100644 --- a/model_search_upload.py +++ b/model_search_upload.py @@ -85,14 +85,14 @@ def uploat_to_mlflow(temp_dir, **context): metrics=['mean_test_score', 'mean_fit_time'] for i, p in enumerate(dct['params'].values()): - with mlflow.start_run(): + with mlflow.start_run(experiment_id=experiment_id): p = json.loads(p.replace('\'', '"')) for parname, parvalue in p.items(): - client.log_param(key=parname, value=parvalue) + mlflow.log_param(key=parname, value=parvalue) for m in metrics: print(f"Logging metric {m} {dct[m][i]}") - client.log_metric(key=m, value=dct[m][i]) + mlflow.log_metric(key=m, value=dct[m][i]) #clean up if 'temp_dir' in attrs: