mirror of
https://github.com/QData/TextAttack.git
synced 2021-10-13 00:05:06 +03:00
47 lines
1.3 KiB
Python
47 lines
1.3 KiB
Python
import torch
|
|
|
|
def get_device():
|
|
return 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 |