explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-12-27 10:26:35 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix get labels for textcat

Browse Source
This commit is contained in:
Matthw Honnibal 2019-10-07 16:50:15 +02:00
parent cbc2cee2c8
commit fa82277398
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
spacy/pipeline/pipes.pyx
View File

@ -1033,8 +1033,7 @@ class TextCategorizer(Pipe):
return 1 return 1
def begin_training(self, get_gold_tuples=lambda: [], pipeline=None, sgd=None, **kwargs): def begin_training(self, get_gold_tuples=lambda: [], pipeline=None, sgd=None, **kwargs):
for raw_text, annots_brackets in get_gold_tuples(): for raw_text, (_, (cats, _2)) in get_gold_tuples():
cats = annots_brackets.pop()
for cat in cats: for cat in cats:
self.add_label(cat) self.add_label(cat)
if self.model is True: if self.model is True: