spaCy/spacy/tests/parser/test_beam_parse.py
2017-07-22 01:48:35 +02:00

11 lines
226 B
Python

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)