diff --git a/hpbandster/core/result.py b/hpbandster/core/result.py index 926b972..9fbd3fe 100644 --- a/hpbandster/core/result.py +++ b/hpbandster/core/result.py @@ -511,7 +511,7 @@ def get_pandas_dataframe(self, budgets=None, loss_fn=lambda r: r.loss): config['budget'] = r.budget all_configs.append(config) - all_losses.append({'loss': r.loss}) + all_losses.append(loss_fn(r)) #df_x = df_x.append(config, ignore_index=True) #df_y = df_y.append({'loss': r.loss}, ignore_index=True)