explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-07-10 16:22:29 +03:00
Code Issues Packages Projects Releases Wiki Activity

Replace TextCatCNN in pipe configurations and templates

Browse Source
This commit is contained in:
Daniël de Kok 2023-12-08 11:03:04 +01:00
parent 5fa33ce8ce
commit 09800bd7c6
3 changed files with 16 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
spacy/cli/templates/quickstart_training.jinja
View File

@ -279,8 +279,11 @@ no_output_layer = false
{% else -%} {% else -%}
[components.textcat.model] [components.textcat.model]
@architectures = "spacy.TextCatCNN.v2" @architectures = "spacy.TextCatReduce.v1"
exclusive_classes = true exclusive_classes = true
use_reduce_first = false
use_reduce_max = false
use_reduce_mean = true
nO = null nO = null
[components.textcat.model.tok2vec] [components.textcat.model.tok2vec]
@ -317,8 +320,11 @@ no_output_layer = false
{% else -%} {% else -%}
[components.textcat_multilabel.model] [components.textcat_multilabel.model]
@architectures = "spacy.TextCatCNN.v2" @architectures = "spacy.TextCatReduce.v1"
exclusive_classes = false exclusive_classes = false
use_reduce_first = false
use_reduce_max = false
use_reduce_mean = true
nO = null nO = null
[components.textcat_multilabel.model.tok2vec] [components.textcat_multilabel.model.tok2vec]

5
spacy/pipeline/textcat.py
View File

@ -55,8 +55,11 @@ no_output_layer = false
single_label_cnn_config = """ single_label_cnn_config = """
[model] [model]
@architectures = "spacy.TextCatCNN.v2" @architectures = "spacy.TextCatReduce.v1"
exclusive_classes = true exclusive_classes = true
use_reduce_first = false
use_reduce_max = false
use_reduce_mean = true
[model.tok2vec] [model.tok2vec]
@architectures = "spacy.HashEmbedCNN.v2" @architectures = "spacy.HashEmbedCNN.v2"

5
spacy/pipeline/textcat_multilabel.py
View File

@ -53,8 +53,11 @@ no_output_layer = false
multi_label_cnn_config = """ multi_label_cnn_config = """
[model] [model]
@architectures = "spacy.TextCatCNN.v2" @architectures = "spacy.TextCatReduce.v1"
exclusive_classes = false exclusive_classes = false
use_reduce_first = false
use_reduce_max = false
use_reduce_mean = true
[model.tok2vec] [model.tok2vec]
@architectures = "spacy.HashEmbedCNN.v2" @architectures = "spacy.HashEmbedCNN.v2"