We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7970abf commit d9b1a15Copy full SHA for d9b1a15
src/pytorch_tabular/tabular_model_tuner.py
@@ -409,7 +409,7 @@ def tune(
409
params.update({"trial_id": i})
410
trials.append(params)
411
if verbose:
412
- 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]}")
413
414
trials_df = pd.DataFrame(trials)
415
trials = trials_df.pop("trial_id")
0 commit comments