explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-01-13 18:56:36 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add custom error when evaluation throws a KeyError

Browse Source
This commit is contained in:
svlandeg 2020-10-09 12:05:33 +02:00
parent 7c52def5da
commit 18dfb27985
2 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
spacy/errors.py
View File

@ -456,6 +456,9 @@ class Errors:
"issue tracker: http://github.com/explosion/spaCy/issues") "issue tracker: http://github.com/explosion/spaCy/issues")
# TODO: fix numbering after merging develop into master # TODO: fix numbering after merging develop into master
E900 = ("Could not run the full 'nlp' pipeline for evaluation. If you specified "
"frozen components, make sure they were already trained and initialized. "
"You can also consider moving them to the 'disabled' list instead.")
E901 = ("Failed to remove existing output directory: {path}. If your " E901 = ("Failed to remove existing output directory: {path}. If your "
"config and the components you train change between runs, a " "config and the components you train change between runs, a "
"non-empty output directory can lead to stale pipeline data. To " "non-empty output directory can lead to stale pipeline data. To "

5
spacy/training/loop.py
View File

@ -249,7 +249,10 @@ def create_evaluation_callback(
def evaluate() -> Tuple[float, Dict[str, float]]: def evaluate() -> Tuple[float, Dict[str, float]]:
dev_examples = list(dev_corpus(nlp)) dev_examples = list(dev_corpus(nlp))
scores = nlp.evaluate(dev_examples) try:
scores = nlp.evaluate(dev_examples)
except KeyError as e:
raise KeyError(Errors.E900) from e
# Calculate a weighted sum based on score_weights for the main score. # Calculate a weighted sum based on score_weights for the main score.
# We can only consider scores that are ints/floats, not dicts like # We can only consider scores that are ints/floats, not dicts like
# entity scores per type etc. # entity scores per type etc.