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 process

Browse Source
This commit is contained in:
Richard Hudson 2021-11-29 21:12:11 +01:00
parent 2d0e916220
commit a660a7d347
1 changed files with 29 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
spacy/visualization.py
View File

@ -37,6 +37,7 @@ ROOT_LEFT_CHARS = {
16: ">", 16: ">",
} }
class Visualizer: class Visualizer:
@staticmethod @staticmethod
def render_dependency_trees(doc: Doc) -> list[str]: def render_dependency_trees(doc: Doc) -> list[str]:
@ -147,20 +148,25 @@ class Visualizer:
2 * horizontal_line_lengths[working_token_index] 2 * horizontal_line_lengths[working_token_index]
) )
# Draw the corners of the relation
char_matrix[first_index_in_relation][char_horizontal_line_length - 1] |= ( char_matrix[first_index_in_relation][char_horizontal_line_length - 1] |= (
HORIZONTAL_LINE_PART_AWAY_FROM_ROOT + LOWER_VERTICAL_LINE_PART HORIZONTAL_LINE_PART_AWAY_FROM_ROOT + LOWER_VERTICAL_LINE_PART
) )
char_matrix[second_index_in_relation][char_horizontal_line_length - 1] |= ( char_matrix[second_index_in_relation][char_horizontal_line_length - 1] |= (
HORIZONTAL_LINE_PART_TOWARDS_ROOT + UPPER_VERTICAL_LINE_PART HORIZONTAL_LINE_PART_AWAY_FROM_ROOT + UPPER_VERTICAL_LINE_PART
) )
# Draw the horizontal line for the governing token
for working_horizontal_position in range(char_horizontal_line_length - 1): for working_horizontal_position in range(char_horizontal_line_length - 1):
if ( if (
char_matrix[working_token_index][working_horizontal_position] char_matrix[head_token_index][working_horizontal_position]
!= VERTICAL_LINE != VERTICAL_LINE
): ):
char_matrix[working_token_index][ char_matrix[head_token_index][
working_horizontal_position working_horizontal_position
] |= HORIZONTAL_LINE ] |= HORIZONTAL_LINE
# Draw the vertical line for the relation
for working_vertical_position in range( for working_vertical_position in range(
first_index_in_relation + 1, second_index_in_relation first_index_in_relation + 1, second_index_in_relation
): ):
@ -175,41 +181,45 @@ class Visualizer:
] |= VERTICAL_LINE ] |= VERTICAL_LINE
for working_token_index in (i for i in range(len(doc)) if heads[i] is not None): for working_token_index in (i for i in range(len(doc)) if heads[i] is not None):
for working_horizontal_position in range( for working_horizontal_position in range(
2 * horizontal_line_lengths[working_token_index], -1, -1 2 * horizontal_line_lengths[working_token_index] - 2, -1, -1
): ):
print(working_token_index, working_horizontal_position)
if ( if (
working_horizontal_position > 0 char_matrix[working_token_index][working_horizontal_position]
and char_matrix[working_token_index][working_horizontal_position] == VERTICAL_LINE) and working_horizontal_position > 1 and char_matrix[working_token_index][working_horizontal_position - 2] == SPACE:
== VERTICAL_LINE # Cross over the existing vertical line, which is owing to a non-projective tree
and char_matrix[ continue
working_token_index][working_horizontal_position - 1
]
== 0
):
# jump over crossing line in non-projective parse case
continue
if ( if (
char_matrix[working_token_index][working_horizontal_position] char_matrix[working_token_index][working_horizontal_position]
!= SPACE != SPACE
): ):
# Draw the arrowhead to the right of what is already there
char_matrix[working_token_index][ char_matrix[working_token_index][
working_horizontal_position + 1 working_horizontal_position + 1
] = ARROWHEAD ] = ARROWHEAD
break break
if working_horizontal_position == 0: if working_horizontal_position == 0:
# Draw the arrowhead at the boundary of the diagram
char_matrix[working_token_index][ char_matrix[working_token_index][
working_horizontal_position working_horizontal_position] = ARROWHEAD
] = ARROWHEAD
else: else:
# Fill in the horizontal line for the governed token
char_matrix[working_token_index][ char_matrix[working_token_index][
working_horizontal_position working_horizontal_position
] |= HORIZONTAL_LINE ] |= HORIZONTAL_LINE
return [''.join(ROOT_RIGHT_CHARS[char_matrix[vertical_position][horizontal_position]] for horizontal_position in range((max_horizontal_line_length * 2 - 1))) for vertical_position in range(len(doc))] return [
"".join(
ROOT_RIGHT_CHARS[char_matrix[vertical_position][horizontal_position]]
for horizontal_position in range((max_horizontal_line_length * 2))
)
for vertical_position in range(len(doc))
]
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")
print (Visualizer().render_dependency_trees(doc)) lines = Visualizer().render_dependency_trees(doc)
for line in lines:
print(line)