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

don't keep recomputing self._label_map for each span

Browse Source
This commit is contained in:
kadarakos 2023-03-03 15:41:41 +00:00
parent c7e7343999
commit 97fd9741c6
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/pipeline/spancat.py
View File

@ -567,6 +567,7 @@ class SpanCategorizer(TrainablePipe):
if self.add_negative_label: if self.add_negative_label:
negative_spans = numpy.ones((scores.shape[0])) negative_spans = numpy.ones((scores.shape[0]))
offset = 0 offset = 0
label_map = self._label_map
for i, eg in enumerate(examples): for i, eg in enumerate(examples):
# Map (start, end) offset of spans to the row in the d_scores array, # Map (start, end) offset of spans to the row in the d_scores array,
# so that we can adjust the gradient for predictions that were # so that we can adjust the gradient for predictions that were
@ -581,7 +582,7 @@ class SpanCategorizer(TrainablePipe):
key = (gold_span.start, gold_span.end) key = (gold_span.start, gold_span.end)
if key in spans_index: if key in spans_index:
row = spans_index[key] row = spans_index[key]
k = self._label_map[gold_span.label_] k = label_map[gold_span.label_]
target[row, k] = 1.0 target[row, k] = 1.0
if self.add_negative_label: if self.add_negative_label:
# delete negative label target. # delete negative label target.