diff --git a/dan/ocr/utils.py b/dan/ocr/utils.py index 988ff985f81c0f0f18bfd31d30fb8e348e2edc19..cb6fe6e97ab7f60aaa171ad7e75453f70505ac50 100644 --- a/dan/ocr/utils.py +++ b/dan/ocr/utils.py @@ -93,6 +93,7 @@ def add_metrics_table_row( continue metric_name = REVERSE_HEADER[column] - row.append(metrics.get(metric_name, "−")) + metric_value = metrics.get(metric_name) + row.append(round(metric_value * 100, 2) if metric_value is not None else "−") table.add_row(row) diff --git a/tests/data/evaluate/metrics_table.md b/tests/data/evaluate/metrics_table.md index e9cc598fe4875bb8619ed08af9feef204eb92e8b..0ff41bfbbd7e9b6736e29d967cb7cfb536120ba9 100644 --- a/tests/data/evaluate/metrics_table.md +++ b/tests/data/evaluate/metrics_table.md @@ -1,5 +1,5 @@ | Split | CER (HTR-NER) | CER (HTR) | WER (HTR-NER) | WER (HTR) | WER (HTR no punct) | |:-----:|:-------------:|:---------:|:-------------:|:---------:|:------------------:| -| train | 1.3023 | 1.3023 | 1.0 | 1.0 | 1.0 | -| val | 1.2683 | 1.2683 | 1.0 | 1.0 | 1.0 | -| test | 1.1224 | 1.1224 | 1.0 | 1.0 | 1.0 | +| train | 130.23 | 130.23 | 100.0 | 100.0 | 100.0 | +| val | 126.83 | 126.83 | 100.0 | 100.0 | 100.0 | +| test | 112.24 | 112.24 | 100.0 | 100.0 | 100.0 | diff --git a/tests/test_evaluate.py b/tests/test_evaluate.py index 425f85ec915adb905f32d761e4c55bb8f47815d4..06958d64eaccdaf6d587a17cbf9535ca4b1ea87f 100644 --- a/tests/test_evaluate.py +++ b/tests/test_evaluate.py @@ -44,7 +44,7 @@ def test_add_metrics_table_row(): "WER (HTR-NER)", "NER", ] - assert metrics_table.rows == [["train", 1.3023, 1.0, "−"]] + assert metrics_table.rows == [["train", 130.23, 100, "−"]] @pytest.mark.parametrize(