From bcd71def6cea5e500aeb1b64875e003464d2fbdd Mon Sep 17 00:00:00 2001
From: Suhan Cui <51844791+SH-Src@users.noreply.github.com>
Date: Tue, 30 Jul 2024 12:14:48 +0800
Subject: [PATCH] Dm (#132)

---
 .../scenarios/data_mining/experiment/model_template/train.py | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/rdagent/scenarios/data_mining/experiment/model_template/train.py b/rdagent/scenarios/data_mining/experiment/model_template/train.py
index f42fa347..26b343ff 100644
--- a/rdagent/scenarios/data_mining/experiment/model_template/train.py
+++ b/rdagent/scenarios/data_mining/experiment/model_template/train.py
@@ -96,8 +96,5 @@ acc = roc_auc_score(y_test, np.concatenate(y_pred))
 
 print(acc)
 
-res = pd.Series(data=[acc], index=['AUROC'])
+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))