diff options
author | John Bauer <horatio@gmail.com> | 2022-08-27 10:25:45 +0300 |
---|---|---|
committer | John Bauer <horatio@gmail.com> | 2022-08-27 10:25:45 +0300 |
commit | 2fc1003531171b465fe9027cc3614ab6c6290943 (patch) | |
tree | 58257f97f1db991b35bfd0665c49bc361e949f65 | |
parent | 4c1bd284d6668cf7ca97e17b42b4a2aa50518be6 (diff) |
simple test of macro f1 result in confusion matrix
-rw-r--r-- | stanza/tests/common/test_confusion.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/stanza/tests/common/test_confusion.py b/stanza/tests/common/test_confusion.py index 85ce42c0..9c3df7a7 100644 --- a/stanza/tests/common/test_confusion.py +++ b/stanza/tests/common/test_confusion.py @@ -5,7 +5,7 @@ Test a couple simple confusion matrices and output formats from collections import defaultdict import pytest -from stanza.utils.confusion import format_confusion +from stanza.utils.confusion import format_confusion, confusion_to_macro_f1 pytestmark = [pytest.mark.travis, pytest.mark.pipeline] @@ -55,3 +55,6 @@ def test_output(simple_confusion, short_confusion): assert EXPECTED_SIMPLE_OUTPUT == format_confusion(simple_confusion) assert EXPECTED_SHORT_OUTPUT == format_confusion(short_confusion) +def test_macro_f1(simple_confusion, short_confusion): + assert confusion_to_macro_f1(simple_confusion) == pytest.approx(0.466666666666) + assert confusion_to_macro_f1(short_confusion) == pytest.approx(0.277777777777) |