mirror of
https://github.com/explosion/spaCy.git
synced 2024-11-11 12:18:04 +03:00
Add test for beam parsing
This commit is contained in:
parent
c86445bdfd
commit
28244df4da
10
spacy/tests/parser/test_beam_parse.py
Normal file
10
spacy/tests/parser/test_beam_parse.py
Normal file
|
@ -0,0 +1,10 @@
|
|||
import spacy
|
||||
import pytest
|
||||
|
||||
@pytest.mark.models
|
||||
def test_beam_parse():
|
||||
nlp = spacy.load('en_core_web_sm')
|
||||
doc = nlp(u'Australia is a country', disable=['ner'])
|
||||
ents = nlp.entity(doc, beam_width=2)
|
||||
print(ents)
|
||||
|
Loading…
Reference in New Issue
Block a user