From 626cf824526b00b18043315322f86789278939fa Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Tue, 27 Sep 2022 23:16:33 +0200
Subject: Refactor metrics

---
 text_recognizer/models/metrics/__init__.py |  0
 text_recognizer/models/metrics/cer.py      | 23 -----------------------
 text_recognizer/models/metrics/wer.py      | 23 -----------------------
 3 files changed, 46 deletions(-)
 delete mode 100644 text_recognizer/models/metrics/__init__.py
 delete mode 100644 text_recognizer/models/metrics/cer.py
 delete mode 100644 text_recognizer/models/metrics/wer.py

(limited to 'text_recognizer/models/metrics')

diff --git a/text_recognizer/models/metrics/__init__.py b/text_recognizer/models/metrics/__init__.py
deleted file mode 100644
index e69de29..0000000
diff --git a/text_recognizer/models/metrics/cer.py b/text_recognizer/models/metrics/cer.py
deleted file mode 100644
index 238ecc3..0000000
--- a/text_recognizer/models/metrics/cer.py
+++ /dev/null
@@ -1,23 +0,0 @@
-"""Character Error Rate (CER)."""
-from typing import Sequence
-
-import torch
-import torchmetrics
-
-
-class CharacterErrorRate(torchmetrics.CharErrorRate):
-    """Character error rate metric, allowing for tokens to be ignored."""
-
-    def __init__(self, ignore_tokens: Sequence[int], *args):
-        super().__init__(*args)
-        self.ignore_tokens = set(ignore_tokens)
-
-    def update(self, preds: torch.Tensor, targets: torch.Tensor) -> None:
-        preds_l = [
-            [t for t in pred if t not in self.ignore_tokens] for pred in preds.tolist()
-        ]
-        targets_l = [
-            [t for t in target if t not in self.ignore_tokens]
-            for target in targets.tolist()
-        ]
-        super().update(preds_l, targets_l)
diff --git a/text_recognizer/models/metrics/wer.py b/text_recognizer/models/metrics/wer.py
deleted file mode 100644
index 78f5854..0000000
--- a/text_recognizer/models/metrics/wer.py
+++ /dev/null
@@ -1,23 +0,0 @@
-"""Character Error Rate (CER)."""
-from typing import Sequence
-
-import torch
-import torchmetrics
-
-
-class WordErrorRate(torchmetrics.WordErrorRate):
-    """Character error rate metric, allowing for tokens to be ignored."""
-
-    def __init__(self, ignore_tokens: Sequence[int], *args):
-        super().__init__(*args)
-        self.ignore_tokens = set(ignore_tokens)
-
-    def update(self, preds: torch.Tensor, targets: torch.Tensor) -> None:
-        preds_l = [
-            [t for t in pred if t not in self.ignore_tokens] for pred in preds.tolist()
-        ]
-        targets_l = [
-            [t for t in target if t not in self.ignore_tokens]
-            for target in targets.tolist()
-        ]
-        super().update(preds_l, targets_l)
-- 
cgit v1.2.3-70-g09d2