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

headers changed

This commit is contained in:
Lorenzo Volpi 2023-09-26 14:25:02 +02:00
parent d455c363dd
commit 7575b95a96
2 changed files with 12 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
quacc/baseline.py
View File

@ -72,7 +72,7 @@ def atc_mc(
cols = [
("base", "F"),
("base", "T"),
("atc_mc", "accuracy"),
("atc mc", "accuracy"),
]
results = []
for test in protocol():
@ -112,7 +112,7 @@ def atc_ne(
cols = [
("base", "F"),
("base", "T"),
("atc_ne", "accuracy"),
("atc ne", "accuracy"),
]
results = []
for test in protocol():
@ -165,7 +165,7 @@ def doc_feat(
cols = [
("base", "F"),
("base", "T"),
("doc_feat", "score"),
("doc feat", "score"),
]
results = []
for test in protocol():

18
quacc/evaluation.py
View File

@ -101,11 +101,11 @@ def evaluation_report(
error_cols = []
for err in error_names:
if err == "mae":
error_cols.extend(["mae_estim", "mae_true"])
error_cols.extend(["mae estim", "mae true"])
elif err == "f1":
error_cols.extend(["f1_estim", "f1_true"])
error_cols.extend(["f1 estim", "f1 true"])
elif err == "f1e":
error_cols.extend(["f1e_estim", "f1e_true"])
error_cols.extend(["f1e estim", "f1e true"])
else:
error_cols.append(err)
@ -129,12 +129,12 @@ def evaluation_report(
for err in error_cols:
error_funcs = {
"mae_true": lambda: error.mae(true_prev),
"mae_estim": lambda: error.mae(estim_prev),
"f1_true": lambda: error.f1(true_prev),
"f1_estim": lambda: error.f1(estim_prev),
"f1e_true": lambda: error.f1e(true_prev),
"f1e_estim": lambda: error.f1e(estim_prev),
"mae true": lambda: error.mae(true_prev),
"mae estim": lambda: error.mae(estim_prev),
"f1 true": lambda: error.f1(true_prev),
"f1 estim": lambda: error.f1(estim_prev),
"f1e true": lambda: error.f1e(true_prev),
"f1e estim": lambda: error.f1e(estim_prev),
}
series[("errors", err)] = error_funcs[err]()