Skip to content

Commit aa9b53f

Browse files
danieldkjikanter
authored andcommitted
Add spacy.TextCatParametricAttention.v1 (explosion#13201)
* Add spacy.TextCatParametricAttention.v1 This layer provides is a simplification of the ensemble classifier that only uses paramteric attention. We have found empirically that with a sufficient amount of training data, using the ensemble classifier with BoW does not provide significant improvement in classifier accuracy. However, plugging in a BoW classifier does reduce GPU training and inference performance substantially, since it uses a GPU-only kernel. * Fix merge fallout
1 parent cc289a8 commit aa9b53f

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

pyproject.toml

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,8 +5,9 @@ requires = [
55
"cymem>=2.0.2,<2.1.0",
66
"preshed>=3.0.2,<3.1.0",
77
"murmurhash>=0.28.0,<1.1.0",
8-
"thinc>=9.0.0.dev4,<9.1.0",
9-
"numpy>=1.15.0",
8+
"thinc>=8.2.2,<8.3.0",
9+
"numpy>=1.15.0; python_version < '3.9'",
10+
"numpy>=1.25.0; python_version >= '3.9'",
1011
]
1112
build-backend = "setuptools.build_meta"
1213

requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ spacy-legacy>=4.0.0.dev1,<4.1.0
33
spacy-loggers>=1.0.0,<2.0.0
44
cymem>=2.0.2,<2.1.0
55
preshed>=3.0.2,<3.1.0
6-
thinc>=9.0.0.dev4,<9.1.0
6+
thinc>=8.2.2,<8.3.0
77
ml_datasets>=0.2.0,<0.3.0
88
murmurhash>=0.28.0,<1.1.0
99
wasabi>=0.9.1,<1.2.0

0 commit comments

Comments
 (0)