explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-05-31 19:23:05 +03:00
Code Issues Packages Projects Releases Wiki Activity

make label mapper private

Browse Source
This commit is contained in:
kadarakos 2023-02-20 17:02:27 +00:00
parent 813b3551ed
commit 86d3e78c64
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
spacy/pipeline/spancat.py
View File

@ -425,7 +425,7 @@ class SpanCategorizer(TrainablePipe):
return list(self.labels)
@property
def label_map(self) -> Dict[str, int]:
def _label_map(self) -> Dict[str, int]:
"""RETURNS (Dict[str, int]): The label map."""
return {label: i for i, label in enumerate(self.labels)}
@ -581,7 +581,7 @@ class SpanCategorizer(TrainablePipe):
key = (gold_span.start, gold_span.end)
if key in spans_index:
row = spans_index[key]
k = self.label_map[gold_span.label_]
k = self._label_map[gold_span.label_]
target[row, k] = 1.0
if self.add_negative_label:
# delete negative label target.