apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-05-05 22:02:38 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix evaluation bug

Browse source
This commit is contained in:
Yizhou Chi 2024-09-14 20:04:09 +08:00
parent f856d768fe
commit 9d2c81a127
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
expo/experimenter/experimenter.py
View file

@ -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"]