explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-08-28 16:04:55 +03:00
Code Issues Packages Projects Releases Wiki Activity

Update spacy/cli/find_threshold.py

Browse Source 
Co-authored-by: Adriane Boyd <adrianeboyd@gmail.com>
This commit is contained in:
Raphael Mitsch 2022-09-29 10:40:30 +02:00 committed by GitHub
parent 65e41a52dd
commit 604c5eab3d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
spacy/cli/find_threshold.py
View File

@ -167,9 +167,7 @@ def find_threshold(
)
scores[threshold] = nlp.evaluate(dev_dataset)[scores_key]
if not isinstance(scores[threshold], float) and not isinstance(
scores[threshold], int
):
if not isinstance(scores[threshold], (float, int)):
wasabi.msg.fail(
f"Returned score for key '{scores_key}' is not numeric. Threshold optimization only works for numeric "
f"scores.",