|
|
"""Visualization utils.""" |
|
|
|
|
|
import numpy as np |
|
|
from IPython.core.display import HTML, display |
|
|
|
|
|
|
|
|
def _get_color(attr): |
|
|
|
|
|
attr = max(-1, min(1, attr)) |
|
|
if attr > 0: |
|
|
hue = 220 |
|
|
sat = 100 |
|
|
lig = 100 - int(90 * attr) |
|
|
else: |
|
|
hue = 220 |
|
|
sat = 100 |
|
|
lig = 100 - int(-125 * attr) |
|
|
return "hsl({}, {}%, {}%)".format(hue, sat, lig) |
|
|
|
|
|
|
|
|
def format_special_tokens(token): |
|
|
"""Convert <> to # if there are any HTML syntax tags. |
|
|
|
|
|
Example: '<Hello>' will be converted to '#Hello' to avoid confusion |
|
|
with HTML tags. |
|
|
|
|
|
Args: |
|
|
token (str): The token to be formatted. |
|
|
Returns: |
|
|
(str): The formatted token. |
|
|
""" |
|
|
if token.startswith("<") and token.endswith(">"): |
|
|
return "#" + token.strip("<>") |
|
|
return token |
|
|
|
|
|
|
|
|
def format_word_importances( |
|
|
words, |
|
|
importances, |
|
|
ground_text_spans, |
|
|
predicted_text_spans, |
|
|
): |
|
|
if np.isnan(importances[0]): |
|
|
importances = np.zeros_like(importances) |
|
|
|
|
|
assert len(words) <= len(importances) |
|
|
tags = ["<div><b>Text</b>: "] |
|
|
|
|
|
for word_index, (word, importance) in enumerate( |
|
|
zip(words, importances[: len(words)]) |
|
|
): |
|
|
word = format_special_tokens(word) |
|
|
for character in word: |
|
|
if ord(character) >= 128: |
|
|
print(word) |
|
|
break |
|
|
color = _get_color(importance) |
|
|
|
|
|
unwrapped_tag = f'<mark style="background-color: {color}; opacity:1.0; \ |
|
|
line-height:1.75"><font color="black"> {word}\ |
|
|
</font></mark>' |
|
|
tags.append(unwrapped_tag) |
|
|
tags.append("</div>") |
|
|
tags.append("<br><span> <b>Ground Spans</b>: [ ") |
|
|
for i, span in enumerate(ground_text_spans): |
|
|
if i != len(ground_text_spans) - 1: |
|
|
tags.append(f"'{span}',") |
|
|
else: |
|
|
tags.append(f"'{span}'") |
|
|
tags.append(" ]</span>") |
|
|
tags.append("<br><span> <b>Predicted Spans</b>: [ ") |
|
|
for i, span in enumerate(predicted_text_spans): |
|
|
if i != len(predicted_text_spans) - 1: |
|
|
tags.append(f"'{span}',") |
|
|
else: |
|
|
tags.append(f"'{span}'") |
|
|
tags.append(" ]</span>") |
|
|
return HTML("".join(tags)) |
|
|
|
|
|
|
|
|
def format_word_colors(words, colors): |
|
|
assert len(words) == len(colors) |
|
|
tags = ["<div style='width:50%;'>"] |
|
|
for word, color in zip(words, colors): |
|
|
word = format_special_tokens(word) |
|
|
unwrapped_tag = '<mark style="background-color: {color}; opacity:1.0; \ |
|
|
line-height:1.75"><font color="black"> {word}\ |
|
|
</font></mark>'.format( |
|
|
color=color, word=word |
|
|
) |
|
|
tags.append(unwrapped_tag) |
|
|
tags.append("</div>") |
|
|
return HTML("".join(tags)) |
|
|
|
|
|
|
|
|
def display_html(html): |
|
|
display(html) |
|
|
|
|
|
|
|
|
def save_to_file(html, path): |
|
|
with open(path, "w") as f: |
|
|
f.write(html.data) |
|
|
|
