From e1d517bd6cd501f1ee0e9840c699a62cac5e75cd Mon Sep 17 00:00:00 2001 From: jj Date: Mon, 5 Feb 2024 12:02:58 +0100 Subject: [PATCH] debug info, probably dislib issue --- model_search_upload.py | 1 + 1 file changed, 1 insertion(+) diff --git a/model_search_upload.py b/model_search_upload.py index 51467ac..6316b18 100644 --- a/model_search_upload.py +++ b/model_search_upload.py @@ -82,6 +82,7 @@ def uploat_to_mlflow(temp_dir, **context): print("Uploading model search results") df = pd.read_csv(os.path.join(temp_dir, 'pd.csv'), index_col=0) dct = df.to_dict() + print("Got following data", dct) metrics=['mean_test_score', 'mean_fit_time'] for i, p in enumerate(dct['params'].values()):