diff options
Diffstat (limited to 'text_recognizer/models/metrics.py')
-rw-r--r-- | text_recognizer/models/metrics.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/text_recognizer/models/metrics.py b/text_recognizer/models/metrics.py index f83c9e4..e59a830 100644 --- a/text_recognizer/models/metrics.py +++ b/text_recognizer/models/metrics.py @@ -1,20 +1,20 @@ """Character Error Rate (CER).""" from typing import Set -import attr +from attrs import define, field import editdistance import torch from torch import Tensor from torchmetrics import Metric -@attr.s(eq=False) +@define(eq=False) class CharacterErrorRate(Metric): """Character error rate metric, computed using Levenshtein distance.""" - ignore_indices: Set[Tensor] = attr.ib(converter=set) - error: Tensor = attr.ib(init=False) - total: Tensor = attr.ib(init=False) + ignore_indices: Set[Tensor] = field(converter=set) + error: Tensor = field(init=False) + total: Tensor = field(init=False) def __attrs_post_init__(self) -> None: super().__init__() |