This commit is contained in:
kadarakos 2023-04-06 13:43:15 +00:00
parent 9cfcbdd0ad
commit 8fc84fb6ab
5 changed files with 28 additions and 25 deletions

View File

@ -1,8 +1,8 @@
from .entity_linker import * # noqa
from .multi_task import * # noqa
from .parser import * # noqa
from .spancat import * # noqa
from .span_finder import * # noqa
from .spancat import * # noqa
from .tagger import * # noqa
from .textcat import * # noqa
from .tok2vec import * # noqa

View File

@ -1,9 +1,11 @@
from typing import Callable, List, Tuple
from thinc.api import Model, chain, with_array
from thinc.types import Floats2d, Floats1d
from spacy.tokens import Doc
from ...util import registry
from thinc.api import Model, chain, with_array
from thinc.types import Floats1d, Floats2d
from spacy.tokens import Doc
from ...util import registry
InT = List[Doc]
OutT = Floats2d

View File

@ -2,22 +2,22 @@ from .attributeruler import AttributeRuler
from .dep_parser import DependencyParser
from .edit_tree_lemmatizer import EditTreeLemmatizer
from .entity_linker import EntityLinker
from .ner import EntityRecognizer
from .entityruler import EntityRuler
from .functions import merge_entities, merge_noun_chunks, merge_subtokens
from .lemmatizer import Lemmatizer
from .morphologizer import Morphologizer
from .ner import EntityRecognizer
from .pipe import Pipe
from .trainable_pipe import TrainablePipe
from .senter import SentenceRecognizer
from .sentencizer import Sentencizer
from .senter import SentenceRecognizer
from .span_finder import SpanFinder
from .span_ruler import SpanRuler
from .spancat import SpanCategorizer
from .tagger import Tagger
from .textcat import TextCategorizer
from .spancat import SpanCategorizer
from .span_ruler import SpanRuler
from .span_finder import SpanFinder
from .textcat_multilabel import MultiLabel_TextCategorizer
from .tok2vec import Tok2Vec
from .functions import merge_entities, merge_noun_chunks, merge_subtokens
from .trainable_pipe import TrainablePipe
__all__ = [
"AttributeRuler",

View File

@ -1,18 +1,19 @@
from typing import List, Dict, Callable, Tuple, Optional, Iterable, Any, cast
from functools import partial
from thinc.api import Config, Model, set_dropout_rate
from thinc.api import Optimizer, get_current_ops, Ops
from thinc.types import Floats2d, Ragged, Ints1d
from typing import Any, Callable, Dict, Iterable, List, Optional, Tuple, cast
from thinc.api import (Config, Model, Ops, Optimizer, get_current_ops,
set_dropout_rate)
from thinc.types import Floats2d, Ints1d, Ragged
from spacy.language import Language
from spacy.pipeline.trainable_pipe import TrainablePipe
from spacy.scorer import Scorer
from spacy.tokens import Doc
from spacy.training import Example
from spacy.scorer import Scorer
from ..util import registry
from .spancat import Suggester
span_finder_default_config = """
[model]
@architectures = "spacy.SpanFinder.v1"

View File

@ -1,13 +1,13 @@
from spacy.language import Language
from spacy.util import registry
from spacy.tokens import Doc
from spacy.training import Example
import pytest
from thinc.api import Config
from thinc.types import Ragged
from spacy.pipeline.span_finder import DEFAULT_PREDICTED_KEY
from spacy.pipeline.span_finder import span_finder_default_config
import pytest
from spacy.language import Language
from spacy.pipeline.span_finder import (DEFAULT_PREDICTED_KEY,
span_finder_default_config)
from spacy.tokens import Doc
from spacy.training import Example
from spacy.util import registry
TRAINING_KEY = "pytest"