explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-08-04 12:20:20 +03:00
Code Issues Packages Projects Releases Wiki Activity

fix tests by fixing textcat config

Browse Source
This commit is contained in:
thomashacker 2023-01-26 00:23:30 +01:00
parent 77cf0ee75d
commit f1da4997b9
1 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
spacy/tests/training/test_rehearse.py
View File

@ -173,7 +173,7 @@ ngram_size = 1
no_output_layer = false no_output_layer = false
[components.textcat_multilabel.model.tok2vec] [components.textcat_multilabel.model.tok2vec]
@architectures = "spacy.Tok2VecListener.v1" @architectures = "spacy.Tok2VecListener.v1"
width = 64 width = ${components.tok2vec.model.encode.width}
upstream = "*" upstream = "*"
[components.tok2vec] [components.tok2vec]
factory = "tok2vec" factory = "tok2vec"
@ -181,15 +181,16 @@ factory = "tok2vec"
@architectures = "spacy.Tok2Vec.v2" @architectures = "spacy.Tok2Vec.v2"
[components.tok2vec.model.embed] [components.tok2vec.model.embed]
@architectures = "spacy.MultiHashEmbed.v2" @architectures = "spacy.MultiHashEmbed.v2"
width = 64 width = ${components.tok2vec.model.encode.width}
attrs = ["ORTH", "SHAPE"] attrs = ["NORM", "PREFIX", "SUFFIX", "SHAPE"]
rows = [5000, 2500] rows = [5000, 1000, 2500, 2500]
include_static_vectors = true include_static_vectors = false
[components.tok2vec.model.encode] [components.tok2vec.model.encode]
@architectures = "spacy.MishWindowEncoder.v2" @architectures = "spacy.MaxoutWindowEncoder.v2"
width = 64 width = 256
depth = 4 depth = 8
window_size = 1 window_size = 1
maxout_pieces = 3
""" """
NER_LISTENER_CONFIG = """ NER_LISTENER_CONFIG = """