mirror of
https://github.com/QData/TextAttack.git
synced 2021-10-13 00:05:06 +03:00
49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
import torch
|
|
|
|
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
|
|
|
|
def html_style_from_dict(style_dict):
|
|
""" Turns
|
|
{ 'color': 'red', 'height': '100px'}
|
|
into
|
|
style: "color: red; height: 100px"
|
|
"""
|
|
style_str = ""
|
|
for key in style_dict:
|
|
style_str += key + ": " + style_dict[key] + ";"
|
|
return 'style="{}"'.format(style_str)
|
|
|
|
|
|
def html_table_from_rows(rows, title=None, header=None, style_dict=None):
|
|
# Stylize the container div.
|
|
if style_dict:
|
|
table_html = "<div {}>".format(style_from_dict(style_dict))
|
|
else:
|
|
table_html = "<div>"
|
|
# Print the title string.
|
|
if title:
|
|
table_html += "<h1>{}</h1>".format(title)
|
|
|
|
# Construct each row as HTML.
|
|
table_html = '<table class="table">'
|
|
if header:
|
|
table_html += "<tr>"
|
|
for element in header:
|
|
table_html += "<th>"
|
|
table_html += str(element)
|
|
table_html += "</th>"
|
|
table_html += "</tr>"
|
|
for row in rows:
|
|
table_html += "<tr>"
|
|
for element in row:
|
|
table_html += "<td>"
|
|
table_html += str(element)
|
|
table_html += "</td>"
|
|
table_html += "</tr>"
|
|
|
|
# Close the table and print to screen.
|
|
table_html += "</table></div>"
|
|
|
|
return table_html
|