From 9d2c81a127de86b7ddcdceba6ad0f6db0eac6073 Mon Sep 17 00:00:00 2001 From: Yizhou Chi Date: Sat, 14 Sep 2024 20:04:09 +0800 Subject: [PATCH] fix evaluation bug --- expo/experimenter/experimenter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/expo/experimenter/experimenter.py b/expo/experimenter/experimenter.py index 89d589d7d..b7a0e0b2f 100644 --- a/expo/experimenter/experimenter.py +++ b/expo/experimenter/experimenter.py @@ -87,7 +87,7 @@ class Experimenter: pred_node_path = os.path.join(state["node_dir"], f"{self.start_time}-{split}_predictions.csv") gt_path = os.path.join(state["datasets_dir"][f"{split}_target"]) preds = pd.read_csv(pred_path) - preds = preds[preds.columns.tolist()[0]] + preds = preds[preds.columns.tolist()[-1]] preds.to_csv(pred_node_path, index=False) gt = pd.read_csv(gt_path)["target"] metric = state["dataset_config"]["metric"]