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

Support unseen classes in parser

Browse Source
This commit is contained in:
Matthew Honnibal 2021-10-25 22:26:52 +02:00
parent 4b5d1b53f6
commit 07a3581ff8
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
spacy/pipeline/transition_parser.pyx
View File

@ -267,6 +267,12 @@ class Parser(TrainablePipe):
gZ = exp_gscores.sum(axis=1, keepdims=True)
d_scores = exp_scores / Z
d_scores[is_gold] -= exp_gscores / gZ
if "unseen_classes" in model.attrs:
for i in range(costs.shape[0]):
for clas in range(costs.shape[1]):
if costs[i, clas] <= best_costs[i, 0]:
if clas in model.attrs["unseen_classes"]:
model.attrs["unseen_classes"].remove(clas)
return d_scores
def _get_costs_from_histories(self, examples, histories):