From f2f4636a3371554dc152553d12866e4644eb75a2 Mon Sep 17 00:00:00 2001 From: Suhan Cui <51844791+SH-Src@users.noreply.github.com> Date: Tue, 30 Jul 2024 11:38:02 +0800 Subject: [PATCH] Dm (#131) * template * a --- .../data_mining/experiment/model_template/train.py | 7 +++++-- rdagent/scenarios/data_mining/experiment/workspace.py | 5 ++--- 2 files changed, 7 insertions(+), 5 deletions(-) diff --git a/rdagent/scenarios/data_mining/experiment/model_template/train.py b/rdagent/scenarios/data_mining/experiment/model_template/train.py index 05a670c9..f42fa347 100644 --- a/rdagent/scenarios/data_mining/experiment/model_template/train.py +++ b/rdagent/scenarios/data_mining/experiment/model_template/train.py @@ -95,6 +95,9 @@ for data in test_dataloader: acc = roc_auc_score(y_test, np.concatenate(y_pred)) print(acc) + +res = pd.Series(data=[acc], index=['AUROC']) +res.to_csv("./submission.csv") # Save the predictions to submission.csv -with open("./submission.txt", "w") as f: - f.write(str(acc)) +# with open("./submission.txt", "w") as f: +# f.write(str(acc)) diff --git a/rdagent/scenarios/data_mining/experiment/workspace.py b/rdagent/scenarios/data_mining/experiment/workspace.py index e2e9d3a1..0f9baa6b 100644 --- a/rdagent/scenarios/data_mining/experiment/workspace.py +++ b/rdagent/scenarios/data_mining/experiment/workspace.py @@ -23,10 +23,9 @@ class DMFBWorkspace(FBWorkspace): env=run_env, ) - csv_path = self.workspace_path / "submission.txt" + csv_path = self.workspace_path / "submission.csv" if not csv_path.exists(): logger.error(f"File {csv_path} does not exist.") return None - with open(self.workspace_path / "submission.txt", "r") as f: - return f.read() + return pd.read_csv(csv_path, index_col=0).iloc[:, 0]