diff --git a/src/pytorch_tabular/tabular_model_tuner.py b/src/pytorch_tabular/tabular_model_tuner.py index 6fd5684e..d199d1fb 100644 --- a/src/pytorch_tabular/tabular_model_tuner.py +++ b/src/pytorch_tabular/tabular_model_tuner.py @@ -409,7 +409,7 @@ def tune( params.update({"trial_id": i}) trials.append(params) if verbose: - logger.info(f"Trial {i+1}/{n_trials}: {params} | Score: {params[metric]}") + logger.info(f"Trial {i+1}/{n_trials}: {params} | Score: {params[metric_str]}") trials_df = pd.DataFrame(trials) trials = trials_df.pop("trial_id")