From b32b28eb125136ca404200ff7e20831442d36f90 Mon Sep 17 00:00:00 2001 From: Bangbang Date: Sat, 14 Sep 2024 12:13:55 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E9=A2=84=E6=B5=8B=E7=BB=93?= =?UTF-8?q?=E6=9E=9C=E6=B2=A1=E6=9C=89target=E5=88=97=E5=90=8D=E3=80=82?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- expo/experimenter/experimenter.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/expo/experimenter/experimenter.py b/expo/experimenter/experimenter.py index b1b5a93c0..418e0089a 100644 --- a/expo/experimenter/experimenter.py +++ b/expo/experimenter/experimenter.py @@ -85,7 +85,8 @@ class Experimenter: os.makedirs(state["node_dir"], exist_ok=True) 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)["target"] + preds = pd.read_csv(pred_path) + preds = preds[preds.columns.tolist()[0]] preds.to_csv(pred_node_path, index=False) gt = pd.read_csv(gt_path)["target"] metric = state["dataset_config"]["metric"]