mirror of https://github.com/explosion/spaCy.git
Add test for beam parsing
This commit is contained in:
parent
c86445bdfd
commit
28244df4da
|
@ -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