This commit is contained in:
Daniël de Kok 2024-01-29 11:49:03 +01:00
parent 8d89d58185
commit eba298edea
2 changed files with 5 additions and 5 deletions

View File

@ -28,10 +28,9 @@ from spacy.tokens import Doc, DocBin
from spacy.training import Example
from spacy.training.initialize import init_nlp
from ..util import make_tempdir
# Ensure that the architecture gets added to the registry.
from ..tok2vec import build_lazy_init_tok2vec as _
from ..util import make_tempdir
TRAIN_DATA_SINGLE_LABEL = [
("I'm so happy.", {"cats": {"POSITIVE": 1.0, "NEGATIVE": 0.0}}),

View File

@ -1,9 +1,10 @@
from typing import List
from spacy.util import registry
from thinc.types import Floats2d
from spacy.tokens import Doc
from thinc.api import Model
from thinc.types import Floats2d
from spacy.tokens import Doc
from spacy.util import registry
@registry.architectures("LazyInitTok2Vec.v1")