From 020a7e5d52f580b53e24575212c792f065005518 Mon Sep 17 00:00:00 2001 From: ines Date: Sat, 9 Dec 2017 15:11:12 +0100 Subject: [PATCH] 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). --- spacy/displacy/__init__.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/spacy/displacy/__init__.py b/spacy/displacy/__init__.py index cfecd4b11..e98016e93 100644 --- a/spacy/displacy/__init__.py +++ b/spacy/displacy/__init__.py @@ -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: