mirror of
https://github.com/explosion/spaCy.git
synced 2025-04-28 04:43:42 +03:00
Merge branch 'add/exclusive-spancat' of github.com:ljvmiranda921/spaCy into add/exclusive-spancat
This commit is contained in:
commit
616723e902
|
@ -64,7 +64,7 @@ def make_spancat(
|
||||||
allow_overlap: bool = True,
|
allow_overlap: bool = True,
|
||||||
) -> "Exclusive_SpanCategorizer":
|
) -> "Exclusive_SpanCategorizer":
|
||||||
"""Create a SpanCategorizerExclusive component. The span categorizer consists of two
|
"""Create a SpanCategorizerExclusive component. The span categorizer consists of two
|
||||||
parts: a suggester function that proposes candidate spans, and a labeller
|
parts: a suggester function that proposes candidate spans, and a labeler
|
||||||
model that predicts a single label for each span.
|
model that predicts a single label for each span.
|
||||||
|
|
||||||
suggester (Callable[[Iterable[Doc], Optional[Ops]], Ragged]): A function that suggests spans.
|
suggester (Callable[[Iterable[Doc], Optional[Ops]], Ragged]): A function that suggests spans.
|
||||||
|
@ -81,7 +81,7 @@ def make_spancat(
|
||||||
Scorer.score_spans for the Doc.spans[spans_key] with overlapping
|
Scorer.score_spans for the Doc.spans[spans_key] with overlapping
|
||||||
spans allowed.
|
spans allowed.
|
||||||
negative_weight (float): Multiplier for the loss terms.
|
negative_weight (float): Multiplier for the loss terms.
|
||||||
Can be used to down weigh the negative samples if there are too many.
|
Can be used to downweight the negative samples if there are too many.
|
||||||
allow_overlap (bool): If True the data is assumed to
|
allow_overlap (bool): If True the data is assumed to
|
||||||
contain overlapping spans.
|
contain overlapping spans.
|
||||||
"""
|
"""
|
||||||
|
@ -100,7 +100,7 @@ def make_spancat(
|
||||||
@dataclass
|
@dataclass
|
||||||
class Ranges:
|
class Ranges:
|
||||||
"""
|
"""
|
||||||
Helper class help avoid storing overlapping span.
|
Helper class to avoid storing overlapping spans.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
|
|
|
@ -87,8 +87,8 @@ architectures and their arguments and hyperparameters.
|
||||||
| `threshold` | Minimum probability to consider a prediction positive. Spans with a positive prediction will be saved on the Doc. Defaults to `0.5`. ~~float~~ |
|
| `threshold` | Minimum probability to consider a prediction positive. Spans with a positive prediction will be saved on the Doc. Defaults to `0.5`. ~~float~~ |
|
||||||
| `max_positive` | Maximum number of labels to consider positive per span. Defaults to `None`, indicating no limit. ~~Optional[int]~~ |
|
| `max_positive` | Maximum number of labels to consider positive per span. Defaults to `None`, indicating no limit. ~~Optional[int]~~ |
|
||||||
| `scorer` | The scoring method. Defaults to [`Scorer.score_spans`](/api/scorer#score_spans) for `Doc.spans[spans_key]` with overlapping spans allowed. ~~Optional[Callable]~~ |
|
| `scorer` | The scoring method. Defaults to [`Scorer.score_spans`](/api/scorer#score_spans) for `Doc.spans[spans_key]` with overlapping spans allowed. ~~Optional[Callable]~~ |
|
||||||
| `negative_weight` | Multiplier for the loss terms. It can be used to down weigh the negative samples if there are too many. It is only available when using the `spancat_exclusive` component. ~~float~~ |
|
| `negative_weight` | Multiplier for the loss terms. It can be used to downweight the negative samples if there are too many. It is only available for the `spancat_exclusive` component. ~~float~~ |
|
||||||
| `allow_overlap` | If `True`, the data is assumed to contain overlapping spans. It is only available when using the `spancat_exclusive` component. ~~bool~~ |
|
| `allow_overlap` | If `True`, the data is assumed to contain overlapping spans. It is only available for the `spancat_exclusive` component. ~~bool~~ |
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
@ -106,7 +106,7 @@ architectures and their arguments and hyperparameters.
|
||||||
>
|
>
|
||||||
> ```python
|
> ```python
|
||||||
> # Construction via add_pipe with default model
|
> # Construction via add_pipe with default model
|
||||||
> # Use 'spancat_exclusive' for exclusive clases
|
> # Use 'spancat_exclusive' for exclusive classes
|
||||||
> spancat = nlp.add_pipe("spancat")
|
> spancat = nlp.add_pipe("spancat")
|
||||||
>
|
>
|
||||||
> # Construction via add_pipe with custom model
|
> # Construction via add_pipe with custom model
|
||||||
|
|
Loading…
Reference in New Issue
Block a user