spaCy/website/docs/api/architectures.md

2.2 KiB

title teaser source menu
Model Architectures Pre-defined model architectures included with the core library spacy/ml/models
Tok2Vec
tok2vec
Transformers
transformers
Parser & NER
parser
Text Classification
textcat
Entity Linking
entitylinker

TODO: intro and how architectures work, link to registry, custom models usage etc.

Tok2Vec architectures

spacy.HashEmbedCNN.v1

spacy.HashCharEmbedCNN.v1

spacy.HashCharEmbedBiLSTM.v1

Transformer architectures

spacy-transformers.TransformerModel.v1

Parser & NER architectures

spacy.TransitionBasedParser.v1

Example Config

[model]
@architectures = "spacy.TransitionBasedParser.v1"
nr_feature_tokens = 6
hidden_width = 64
maxout_pieces = 2

[model.tok2vec]
# ...
Name Type Description
tok2vec Model
nr_feature_tokens int
hidden_width int
maxout_pieces int
use_upper bool
nO int

Text classification architectures

spacy.TextCatEnsemble.v1

spacy.TextCatBOW.v1

spacy.TextCatCNN.v1

spacy.TextCatLowData.v1

Entity linking architectures

spacy.EntityLinker.v1