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

Merge branch 'add/exclusive-spancat' of github.com:ljvmiranda921/spaCy into add/exclusive-spancat

Browse Source
This commit is contained in:
Lj Miranda 2022-10-26 11:09:03 +08:00
commit 60a8df7c5f
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
spacy/pipeline/spancat_exclusive.py
View File

@ -1,5 +1,5 @@
from dataclasses import dataclass
from typing import Any, Callable, Dict, Iterable, List, Optional, Tuple, cast
from typing import Callable, Dict, Iterable, List, Optional, Tuple, cast
import numpy
from thinc.api import Config, Model, Ops, Optimizer
@ -370,7 +370,8 @@ class SpanCategorizerExclusive(TrainablePipe):
target[negative_samples, self._negative_label] = 1.0 # type: ignore
d_scores = scores - target
neg_weight = cast(float, self.cfg["negative_weight"])
d_scores[negative_samples] *= neg_weight
if neg_weight != 1.0:
d_scores[negative_samples] *= neg_weight
loss = float((d_scores**2).sum())
return loss, d_scores