Fix #1767

fastai · Mar 7, 2019 · d5c8be7 · d5c8be7
1 parent ad21ce3
commit d5c8be7
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/fastai/metrics.py b/fastai/metrics.py
@@ -217,6 +217,7 @@ def on_epoch_end(self, last_metrics, **kwargs):
 
     def on_train_end(self, **kwargs): self.average = self.avg
 
+@dataclass
 class KappaScore(ConfusionMatrix):
     "Compute the rate of agreement (Cohens Kappa)."
     weights:Optional[str]=None      # None, `linear`, or `quadratic`
@@ -235,7 +236,8 @@ def on_epoch_end(self, last_metrics, **kwargs):
         else: raise ValueError('Unknown weights. Expected None, "linear", or "quadratic".')
         k = torch.sum(w * self.cm) / torch.sum(w * expected)
         return add_metrics(last_metrics, 1-k)
-
+
+@dataclass
 class MatthewsCorreff(ConfusionMatrix):
     "Compute the Matthews correlation coefficient."
     def on_epoch_end(self, last_metrics, **kwargs):