From 30a5421fd766c4646ddb4fb09b7cc7a42d6b9cde Mon Sep 17 00:00:00 2001 From: goulustis Date: Wed, 29 Oct 2025 19:06:28 +0800 Subject: [PATCH] save results --- lang_agent/eval/evaluator.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/lang_agent/eval/evaluator.py b/lang_agent/eval/evaluator.py index c7d07af..9f3b41b 100644 --- a/lang_agent/eval/evaluator.py +++ b/lang_agent/eval/evaluator.py @@ -76,19 +76,21 @@ class Evaluator: exp_save_f = osp.join(f"{head_path}-{n_exp}.csv") df = self.result.to_pandas() + logger.info(f"saving experiment results to: {exp_save_f}") df.to_csv(exp_save_f, index=False) metric_col = [e for e in df.columns if "feedback" in e] - df_curr_m = df[metric_col].mean() + df_curr_m = df[metric_col].mean().to_frame().T + df_curr_m.index = [f'{head_path}-{n_exp}'] metric_f = osp.join(self.config.log_dir, "0_exp_metrics.csv") # start with 0 for first file in folder if osp.exists(metric_f): - df_m = pd.read_csv(metric_f) + df_m = pd.read_csv(metric_f, index_col=0) df_m = pd.concat([df_m, df_curr_m]) else: df_m = df_curr_m - df_m.to_csv(metric_f, index=False) + df_m.to_csv(metric_f)