CI

microsoft · Jul 30, 2024 · b16e4d4 · b16e4d4
1 parent 13c5011
commit b16e4d4
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/rdagent/scenarios/data_mining/experiment/model_template/train.py b/rdagent/scenarios/data_mining/experiment/model_template/train.py
@@ -96,8 +96,5 @@ def collate_fn(batch):
 
 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))