ANL-CEEESA/MIPLearn
1
0
Fork 0
mirror of https://github.com/ANL-CEEESA/MIPLearn.git synced 2025-12-06 01:18:52 -06:00
Code Issues Packages Projects Releases Wiki Activity

Make classifier evaluation metrics work when p=0

Browse Source
This commit is contained in:
Alinson S. Xavier
2020-04-15 10:36:47 -05:00
parent 8039ac7845
commit 5b02c6ea0e
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/python/miplearn/components/__init__.py
View File

@@ -17,9 +17,13 @@ def classifier_evaluation_dict(tp, tn, fp, fn):
"False negative": fn,
"Accuracy": (tp + tn) / (p + n),
"F1 score": (2 * tp) / (2 * tp + fp + fn),
"Recall": tp / p,
"Precision": tp / (tp + fp),
}
if p > 0:
d["Recall"] = tp / p
d["Precision"] = tp / (tp + fp)
else:
d["Recall"] = 1.0
d["Precision"] = 1.0
t = (p + n) / 100.0
d["Predicted positive (%)"] = d["Predicted positive"] / t
d["Predicted negative (%)"] = d["Predicted negative"] / t