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

Fix textcat simple train example

Browse Source
This commit is contained in:
ines 2017-11-07 01:25:54 +01:00
parent 173b1551af
commit 42a0fbf291
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
spacy/tests/pipeline/test_textcat.py
View File

@ -1,19 +1,18 @@
# coding: utf8
from __future__ import unicode_literals from __future__ import unicode_literals
from ...language import Language from ...language import Language
def test_simple_train(): def test_simple_train():
nlp = Language() nlp = Language()
nlp.add_pipe(nlp.create_pipe('textcat')) nlp.add_pipe(nlp.create_pipe('textcat'))
nlp.get_pipe('textcat').add_label('is_good') nlp.get_pipe('textcat').add_label('answer')
nlp.begin_training() nlp.begin_training()
for i in range(5): for i in range(5):
for text, answer in [('aaaa', 1.), ('bbbb', 0), ('aa', 1.), for text, answer in [('aaaa', 1.), ('bbbb', 0), ('aa', 1.),
('bbbbbbbbb', 0.), ('aaaaaa', 1)]: ('bbbbbbbbb', 0.), ('aaaaaa', 1)]:
nlp.update([text], [{'cats': {'answer': answer}}]) nlp.update([text], [{'cats': {'answer': answer}}])
doc = nlp(u'aaa') doc = nlp(u'aaa')
assert 'is_good' in doc.cats assert 'answer' in doc.cats
assert doc.cats['is_good'] >= 0.5 assert doc.cats['answer'] >= 0.5