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 6b9026a commit 26bf642Copy full SHA for 26bf642
2 files changed
spacy/cli/evaluate.py
@@ -175,10 +175,13 @@ def render_parses(
175
def print_prf_per_type(
176
msg: Printer, scores: Dict[str, Dict[str, float]], name: str, type: str
177
) -> None:
178
- data = [
179
- (k, f"{v['p']*100:.2f}", f"{v['r']*100:.2f}", f"{v['f']*100:.2f}")
180
- for k, v in scores.items()
181
- ]
+ data = []
+ for key, value in scores.items():
+ row = [key]
+ for k in ("p", "r", "f"):
182
+ v = value[k]
183
+ row.append(f"{v * 100:.2f}" if isinstance(v, (int, float)) else v)
184
+ data.append(row)
185
msg.table(
186
data,
187
header=("", "P", "R", "F"),
@@ -191,7 +194,10 @@ def print_textcats_auc_per_cat(
191
194
msg: Printer, scores: Dict[str, Dict[str, float]]
192
195
193
196
- [(k, f"{v:.2f}") for k, v in scores.items()],
197
+ [
198
+ (k, f"{v:.2f}" if isinstance(v, (float, int)) else v)
199
+ for k, v in scores.items()
200
+ ],
201
header=("", "ROC AUC"),
202
aligns=("l", "r"),
203
title="Textcat ROC AUC (per label)",
spacy/tests/regression/test_issue7019.py
@@ -0,0 +1,12 @@
1
+from spacy.cli.evaluate import print_textcats_auc_per_cat, print_prf_per_type
2
+from wasabi import msg
3
+
4
5
+def test_issue7019():
6
+ scores = {"LABEL_A": 0.39829102, "LABEL_B": 0.938298329382, "LABEL_C": None}
7
+ print_textcats_auc_per_cat(msg, scores)
8
+ scores = {
9
+ "LABEL_A": {"p": 0.3420302, "r": 0.3929020, "f": 0.49823928932},
10
+ "LABEL_B": {"p": None, "r": None, "f": None},
11
+ }
12
+ print_prf_per_type(msg, scores, name="foo", type="bar")
0 commit comments