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

Fiddle with get_mentions definition

Browse Source 
Ended up not making a difference, but oh well.
This commit is contained in:
Paul O'Leary McCann 2021-05-18 19:53:33 +09:00
parent a33d29441a
commit 051715506e
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/ml/models/coref.py
View File

@ -2,7 +2,7 @@ from dataclasses import dataclass
from thinc.api import Model, Linear, Relu, Dropout, chain, noop from thinc.api import Model, Linear, Relu, Dropout, chain, noop
from thinc.types import Floats2d, Floats1d, Ints2d, Ragged from thinc.types import Floats2d, Floats1d, Ints2d, Ragged
from typing import List, Callable, Tuple from typing import List, Callable, Tuple, Any, Generator
from ...tokens import Doc from ...tokens import Doc
from ...util import registry from ...util import registry
@ -20,7 +20,7 @@ from .coref_util import (
@registry.architectures("spacy.Coref.v0") @registry.architectures("spacy.Coref.v0")
def build_coref( def build_coref(
tok2vec: Model[List[Doc], List[Floats2d]], tok2vec: Model[List[Doc], List[Floats2d]],
get_mentions: Callable = get_candidate_mentions, get_mentions: Any = get_candidate_mentions,
hidden: int = 1000, hidden: int = 1000,
dropout: float = 0.3, dropout: float = 0.3,
mention_limit: int = 3900, mention_limit: int = 3900,