set versions to v1 instead of v0

This commit is contained in:
svlandeg 2021-05-27 16:17:20 +02:00
parent 2e3c0e2256
commit 910026582d
3 changed files with 4 additions and 4 deletions

View File

@ -17,7 +17,7 @@ from .coref_util import (
)
@registry.architectures("spacy.Coref.v0")
@registry.architectures("spacy.Coref.v1")
def build_coref(
tok2vec: Model[List[Doc], List[Floats2d]],
get_mentions: Any = get_candidate_mentions,

View File

@ -148,7 +148,7 @@ def get_candidate_mentions(
return (begins, ends)
@registry.misc("spacy.CorefCandidateGenerator.v0")
@registry.misc("spacy.CorefCandidateGenerator.v1")
def create_mention_generator() -> Any:
return get_candidate_mentions

View File

@ -27,14 +27,14 @@ from ..coref_scorer import Evaluator, get_cluster_info, b_cubed
default_config = """
[model]
@architectures = "spacy.Coref.v0"
@architectures = "spacy.Coref.v1"
max_span_width = 20
mention_limit = 3900
dropout = 0.3
hidden = 1000
[model.get_mentions]
@misc = "spacy.CorefCandidateGenerator.v0"
@misc = "spacy.CorefCandidateGenerator.v1"
[model.tok2vec]
@architectures = "spacy.Tok2Vec.v2"