explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-10-29 23:17:59 +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 a6a2ff217f
commit c8857181f8
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

@ -1032,8 +1032,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: