gavindiaz/NexQuant
1
0
Fork 0
mirror of https://github.com/NicolasBohn/NexQuant.git synced 2026-07-28 07:57:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

Dm (#132)

Browse Source
This commit is contained in:
Suhan Cui
2024-07-30 12:14:48 +08:00
committed by GitHub
parent f2f4636a33
commit bcd71def6c
1 changed files with 1 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/scenarios/data_mining/experiment/model_template/train.py
+1 -4
View File
@@ -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))