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

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

Browse source
This commit is contained in:
Bangbang 2024-09-14 12:13:55 +08:00
parent 7458e43edf
commit b32b28eb12
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"]