From 42a0fbf29168f12a4bc3afc53bbf7148b9d008f6 Mon Sep 17 00:00:00 2001 From: ines Date: Tue, 7 Nov 2017 01:25:54 +0100 Subject: [PATCH] Fix textcat simple train example --- spacy/tests/pipeline/test_textcat.py | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/spacy/tests/pipeline/test_textcat.py b/spacy/tests/pipeline/test_textcat.py index 772131225..99f0f8908 100644 --- a/spacy/tests/pipeline/test_textcat.py +++ b/spacy/tests/pipeline/test_textcat.py @@ -1,19 +1,18 @@ +# coding: utf8 + from __future__ import unicode_literals from ...language import Language + def test_simple_train(): nlp = Language() - 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() - for i in range(5): for text, answer in [('aaaa', 1.), ('bbbb', 0), ('aa', 1.), ('bbbbbbbbb', 0.), ('aaaaaa', 1)]: nlp.update([text], [{'cats': {'answer': answer}}]) doc = nlp(u'aaa') - assert 'is_good' in doc.cats - assert doc.cats['is_good'] >= 0.5 - + assert 'answer' in doc.cats + assert doc.cats['answer'] >= 0.5