From 2482a0e874c6239ffb715c854ff5f7a1e1088168 Mon Sep 17 00:00:00 2001 From: Arne Binder Date: Fri, 1 Dec 2023 11:28:43 +0100 Subject: [PATCH] re-export metrics (and statistics) from pytorch-ie --- src/pie_modules/metrics/__init__.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/src/pie_modules/metrics/__init__.py b/src/pie_modules/metrics/__init__.py index 27081d5b4..33ac23610 100644 --- 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