From e01a5e9961d641ea3db0a9902c7f5c6cb2f492e4 Mon Sep 17 00:00:00 2001 From: leiyre Date: Thu, 10 Mar 2022 15:29:12 +0100 Subject: [PATCH] fix(#1238): show prediction labels when annotating rule (#1239) This PR shows prediction labels when annotating rule in Weak Supervision Closes #1238 * fix(metrics): compute dataset labels as python metric * test: fix tests * fix: compute dataset label properly * fix show all labels for empty query view * empty query * refactor: revert comp. changes and compute labels in model * chore: lint fix Co-authored-by: Francisco Aranda (cherry picked from commit 6f19e405356dda26c6d9d57bff833f6a339ec38d) --- src/rubrix/server/tasks/text_classification/metrics.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/rubrix/server/tasks/text_classification/metrics.py b/src/rubrix/server/tasks/text_classification/metrics.py index f97690d0e6..af7f4406fc 100644 --- a/src/rubrix/server/tasks/text_classification/metrics.py +++ b/src/rubrix/server/tasks/text_classification/metrics.py @@ -4,7 +4,7 @@ from sklearn.metrics import precision_recall_fscore_support from sklearn.preprocessing import MultiLabelBinarizer -from rubrix.server.tasks.commons.metrics import CommonTasksMetrics +from rubrix.server.tasks.commons.metrics import CommonTasksMetrics, GenericRecord from rubrix.server.tasks.commons.metrics.model.base import ( BaseMetric, PythonMetric,