explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-07-14 18:22:27 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add title to dep visualizations

Browse Source
This commit is contained in:
thomashacker 2023-02-14 13:49:13 +01:00
parent ee0be6e484
commit e8f16051b3
3 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
spacy/displacy/render.py
View File

@ -300,6 +300,8 @@ class DependencyRenderer:
self.lang = settings.get("lang", DEFAULT_LANG) self.lang = settings.get("lang", DEFAULT_LANG)
render_id = f"{id_prefix}-{i}" render_id = f"{id_prefix}-{i}"
svg = self.render_svg(render_id, p["words"], p["arcs"]) svg = self.render_svg(render_id, p["words"], p["arcs"])
if p.get("title"):
svg = TPL_TITLE.format(title=p.get("title")) + svg
rendered.append(svg) rendered.append(svg)
if page: if page:
content = "".join([TPL_FIGURE.format(content=svg) for svg in rendered]) content = "".join([TPL_FIGURE.format(content=svg) for svg in rendered])

1
spacy/tests/test_displacy.py
View File

@ -350,6 +350,7 @@ def test_displacy_render_manual_dep():
{"start": 2, "end": 3, "label": "det", "dir": "left"}, {"start": 2, "end": 3, "label": "det", "dir": "left"},
{"start": 1, "end": 3, "label": "attr", "dir": "right"}, {"start": 1, "end": 3, "label": "attr", "dir": "right"},
], ],
"title": "Title",
} }
html = displacy.render([parsed_dep], style="dep", manual=True) html = displacy.render([parsed_dep], style="dep", manual=True)
for word in parsed_dep["words"]: for word in parsed_dep["words"]:

1
website/docs/api/top-level.mdx
View File

@ -371,6 +371,7 @@ displaCy's different data formats below.
| `words` | List of dictionaries describing a word token (see structure below). ~~List[Dict[str, Any]]~~ | | `words` | List of dictionaries describing a word token (see structure below). ~~List[Dict[str, Any]]~~ |
| `arcs` | List of dictionaries describing the relations between words (see structure below). ~~List[Dict[str, Any]]~~ | | `arcs` | List of dictionaries describing the relations between words (see structure below). ~~List[Dict[str, Any]]~~ |
| _Optional_ | | | _Optional_ | |
| `title` | Title of the visualization. ~~Optional[str]~~ |
| `settings` | Dependency Visualizer options (see [here](/api/top-level#displacy_options)). ~~Dict[str, Any]~~ | | `settings` | Dependency Visualizer options (see [here](/api/top-level#displacy_options)). ~~Dict[str, Any]~~ |
<Accordion title="Words data structure"> <Accordion title="Words data structure">