Rooba
/
spaCy
mirror of https://github.com/explosion/spaCy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Allow 'fine_grained' option in displaCy (see #1703) 

Shows token.tag_ instead of token.pos_. Disabled by default, to not cause rendering issues for models with long fine-grained tags (e.g. merged morphological features).
This commit is contained in:
ines 2017-12-09 15:11:12 +01:00
parent d8dd484dc0
commit 020a7e5d52
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
spacy/displacy/__init__.py
View File

@ -97,7 +97,10 @@ def parse_deps(orig_doc, options={}):
word.lemma_, word.ent_type_))
for span_props in spans:
doc.merge(*span_props)
words = [{'text': w.text, 'tag': w.pos_} for w in doc]
if options.get('fine_grained'):
words = [{'text': w.text, 'tag': w.tag_} for w in doc]
else:
words = [{'text': w.text, 'tag': w.pos_} for w in doc]
arcs = []
for word in doc:
if word.i < word.head.i: