We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7ffce00 commit c453251Copy full SHA for c453251
utils/load_metric.py
@@ -4,7 +4,7 @@
4
import torch.nn as nn
5
6
from .metrics import EntropyPrediction
7
-
+from .metrics import precision
8
9
class MetricWrapper(nn.Module):
10
def __init__(self, *metrics):
@@ -39,7 +39,7 @@ def _get_metric(self, key):
39
case "recall":
40
raise NotImplementedError("Recall score not implemented yet")
41
case "precision":
42
- raise NotImplementedError("Precision score not implemented yet")
+ return precision()
43
case "accuracy":
44
raise NotImplementedError("Accuracy score not implemented yet")
45
case _:
0 commit comments