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

Merge branch 'feature/refactor-parser' of github.com:explosion/spaCy into refactor/replace-thinc-linalg-backend

Browse Source
This commit is contained in:
shademe 2022-08-11 13:44:59 +02:00
commit 078dcef442
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
spacy/pipeline/textcat.py
View File

@ -192,7 +192,7 @@ class TextCategorizer(TrainablePipe):
if not any(len(doc) for doc in docs): if not any(len(doc) for doc in docs):
# Handle cases where there are no tokens in any docs. # Handle cases where there are no tokens in any docs.
tensors = [doc.tensor for doc in docs] tensors = [doc.tensor for doc in docs]
xp = get_array_module(tensors) xp = self.model.ops.xp
scores = xp.zeros((len(list(docs)), len(self.labels))) scores = xp.zeros((len(list(docs)), len(self.labels)))
return scores return scores
scores = self.model.predict(docs) scores = self.model.predict(docs)