lorenzo.volpi
/
QuAcc
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

f1 updated

This commit is contained in:
Lorenzo Volpi 2023-12-21 16:47:20 +01:00
parent e01006e663
commit 5d0ecfda39
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
baselines/atc.py
View File

@ -38,7 +38,7 @@ def get_ATC_acc(thres, scores):
return np.mean(scores >= thres)
def get_ATC_f1(thres, scores, probs):
def get_ATC_f1(thres, scores, probs, average="binary"):
preds = np.argmax(probs, axis=-1)
estim_y = np.abs(1 - (scores >= thres) ^ preds)
return f1_score(estim_y, preds)
return f1_score(estim_y, preds, average=average)