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

Merge branch 'fix_evaluate_prediction' into 'expo'

Browse source 
修复预测结果没有target列名。

See merge request agents/exp_optimizer!14
This commit is contained in:
林义章 2024-09-14 04:19:18 +00:00
commit 63bb972f91
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
expo/experimenter/experimenter.py
View file

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