explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-12-26 18:06:29 +03:00
Code Issues Packages Projects Releases Wiki Activity

Save raw scores in evaluate output

Browse Source
This commit is contained in:
Adriane Boyd 2020-10-19 15:03:19 +02:00
parent dd207ca6d0
commit 563a21834e
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
spacy/cli/evaluate.py
View File

@ -93,6 +93,7 @@ def evaluate(
"SPEED": "speed", "SPEED": "speed",
} }
results = {} results = {}
data = {}
for metric, key in metrics.items(): for metric, key in metrics.items():
if key in scores: if key in scores:
if key == "cats_score": if key == "cats_score":
@ -101,7 +102,7 @@ def evaluate(
results[metric] = f"{scores[key]:.0f}" results[metric] = f"{scores[key]:.0f}"
else: else:
results[metric] = f"{scores[key]*100:.2f}" results[metric] = f"{scores[key]*100:.2f}"
data = {re.sub(r"[\s/]", "_", k.lower()): v for k, v in results.items()} data[re.sub(r"[\s/]", "_", key.lower())] = scores[key]
msg.table(results, title="Results") msg.table(results, title="Results")