Merge pull request #13 from ArneBinder/re-export-pytorch_ie-metrics

re-export `metrics` (and `statistics`) from `pytorch-ie`
ArneBinder · Dec 1, 2023 · 15e4414 · 15e4414
2 parents e9eb749 + 2482a0e
commit 15e4414
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/pie_modules/metrics/__init__.py b/src/pie_modules/metrics/__init__.py
@@ -1 +1,9 @@
+from pytorch_ie.metrics import F1Metric
+from pytorch_ie.metrics.statistics import (
+    FieldLengthCollector,
+    LabelCountCollector,
+    SubFieldLengthCollector,
+    TokenCountCollector,
+)
+
 from .span_length_collector import SpanLengthCollector