Skip to content

Commit

Permalink
fix a bug
Browse files Browse the repository at this point in the history
  • Loading branch information
SakiRinn committed Jun 30, 2023
1 parent 780a9d1 commit 6ee81db
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions main.py
Original file line number Diff line number Diff line change
Expand Up @@ -121,9 +121,9 @@ def main():

with open(result_file, "a") as f:
txt = (
f"{args.dataset}\t\t{auc_lst[i]:.4f}\t\tNull\t\t"
f"{ap_lst[i]:.4f}\t\tNull\t\t{f1_lst[i]:.4f}\t\tNull\t\t"
f"{avg_time1:.1f}/{avg_time2:.1f}"
f"{args.dataset:4}\t\t{auc_lst[i]:.4f}\t\tNull\t"
f"{ap_lst[i]:.4f}\t\tNull\t{f1_lst[i]:.4f}\tNull\t"
f"{t1_lst[i]:.1f}/{t2_lst[i]:.1f}"
)
print(txt)
print(txt, file=f)
Expand All @@ -139,8 +139,8 @@ def main():

with open(result_file, "a") as f:
txt = (
f"{args.dataset}\t\t{avg_auc:.4f}\t\t{std_auc:.4f}\t\t"
f"{avg_ap:.4f}\t\t{std_ap:.4f}\t\t{avg_f1:.4f}\t\t{std_f1:.4f}\t\t"
f"{args.dataset:4}\t\t{avg_auc:.4f}\t\t{std_auc:.4f}\t"
f"{avg_ap:.4f}\t\t{std_ap:.4f}\t{avg_f1:.4f}\t{std_f1:.4f}\t"
f"{avg_time1:.1f}/{avg_time2:.1f}"
)
print(txt, file=f)
Expand Down

0 comments on commit 6ee81db

Please sign in to comment.