explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-08-04 12:20:20 +03:00
Code Issues Packages Projects Releases Wiki Activity

Work in progress

Browse Source
This commit is contained in:
Richard Hudson 2021-11-26 22:14:49 +01:00
parent 9d97a0ff0c
commit 93a4905b25
1 changed files with 21 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
spacy/visualization.py
View File

@ -57,14 +57,17 @@ class Visualizer:
heads[working_token_index], heads[working_token_index],
) )
if len(children_lists[working_token_index]) > 0: if len(children_lists[working_token_index]) > 0:
working_horizontal_arrow_position = max( working_horizontal_line_length = (
[ max(
horizontal_line_lengths[i] [
for i in children_lists[working_token_index] horizontal_line_lengths[i]
] for i in children_lists[working_token_index]
]
)
+ 1
) )
else: else:
working_horizontal_arrow_position = 0 working_horizontal_line_length = 1
for inbetween_index in ( for inbetween_index in (
i i
for i in range( for i in range(
@ -73,30 +76,30 @@ class Visualizer:
if i not in children_lists[working_token_index] if i not in children_lists[working_token_index]
and horizontal_line_lengths[i] != 0 and horizontal_line_lengths[i] != 0
): ):
working_horizontal_arrow_position = max( working_horizontal_line_length = max(
working_horizontal_arrow_position, working_horizontal_line_length,
horizontal_line_lengths[inbetween_index] - 1 horizontal_line_lengths[inbetween_index]
if inbetween_index in children_lists[heads[working_token_index]] if inbetween_index in children_lists[heads[working_token_index]]
else horizontal_line_lengths[inbetween_index], else horizontal_line_lengths[inbetween_index] + 1,
) )
for child_horizontal_arrow_position in ( for child_horizontal_line_length in (
horizontal_line_lengths[i] horizontal_line_lengths[i]
for i in children_lists[working_token_index] for i in children_lists[working_token_index]
if (i < first_index_in_relation or i > second_index_in_relation) if (i < first_index_in_relation or i > second_index_in_relation)
and horizontal_line_lengths[i] != 0 and horizontal_line_lengths[i] != 0
): ):
working_horizontal_arrow_position = max( working_horizontal_line_length = max(
working_horizontal_arrow_position, working_horizontal_line_length,
child_horizontal_arrow_position, child_horizontal_line_length + 1,
) )
horizontal_line_lengths[working_token_index] = ( horizontal_line_lengths[
working_horizontal_arrow_position + 1 working_token_index
) ] = working_horizontal_line_length
print(horizontal_line_lengths) print(horizontal_line_lengths)
import spacy import spacy
nlp = spacy.load("en_core_web_sm") nlp = spacy.load("en_core_web_sm")
doc = nlp("I saw a horse yesterday that was injured.") doc = nlp("I saw a horse yesterday that was injured")
Visualizer().render_dependency_trees(doc) Visualizer().render_dependency_trees(doc)