explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-01-12 18:26:30 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add extra beam parsing test

Browse Source
This commit is contained in:
Matthew Honnibal 2017-11-05 14:39:57 +01:00
parent e777ea25bb
commit 00435d8f0c
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
spacy/tests/parser/test_beam_parse.py
View File

@ -2,10 +2,22 @@
from __future__ import unicode_literals from __future__ import unicode_literals
import pytest import pytest
from ...language import Language
from ...pipeline import DependencyParser
@pytest.mark.models('en') @pytest.mark.models('en')
def test_beam_parse(EN): def test_beam_parse_en(EN):
doc = EN(u'Australia is a country', disable=['ner']) doc = EN(u'Australia is a country', disable=['ner'])
ents = EN.entity(doc, beam_width=2) ents = EN.entity(doc, beam_width=2)
print(ents) print(ents)
def test_beam_parse():
nlp = Language()
nlp.add_pipe(DependencyParser(nlp.vocab), name='parser')
nlp.parser.add_label('nsubj')
nlp.begin_training()
doc = nlp.make_doc(u'Australia is a country')
nlp.parser(doc, beam_width=2)