Function convert_to_html_table

benchmarks/benchmark.py:52–105

(results, param_name, main_title=None, comments=None)

def convert_to_html_table(results, param_name, main_title=None, comments=None):
    string = "<table>\n"
    keys = list(results.keys())
    params, names, devices, bitsizes = zip(*keys)

    devices_names = sorted(list(set(zip(devices, names))))
    params = sorted(list(set(params)))
    bitsizes = sorted(list(set(bitsizes)))
    length = len(devices_names) + 1
    cpus_cols = list(devices).count("CPU") / len(bitsizes) / len(params)
    gpus_cols = list(devices).count("GPU") / len(bitsizes) / len(params)
    assert cpus_cols + gpus_cols == len(devices_names)

    if main_title is not None:
        string += (
            f'<tr><th align="center" colspan="{length}">{str(main_title)}</th></tr>\n'
        )

    for i, bitsize in enumerate(bitsizes):
        if i != 0:
            string += f'<tr><td colspan="{length}">&nbsp;</td></tr>\n'

        # make bitsize header
        text = f"{bitsize} bits"
        if comments is not None:
            text += " - "
            if isinstance(comments, (tuple, list)) and len(comments) == len(bitsizes):
                text += str(comments[i])
            else:
                text += str(comments)
        string += f'<tr><th align="center">Bitsize</th>'
        string += f'<th align="center" colspan="{length - 1}">{text}</th></tr>\n'

        # make device header
        string += f'<tr><th align="center">Device</th>'
        string += f'<th align="center" colspan="{cpus_cols}">CPU</th>'
        string += f'<th align="center" colspan="{gpus_cols}">GPU</th></tr>\n'

        # make param_name / backend header
        string += f'<tr><th align="center">{param_name}</th>'
        for device, name in devices_names:
            string += f'<th align="center">{name}</th>'
        string += "</tr>\n"

        # make results rows
        for param in params:
            string += f'<tr><td align="center">{param}</td>'
            for device, name in devices_names:
                key = (param, name, device, bitsize)
                string += f'<td align="center">{results[key]:.4f}</td>'
            string += "</tr>\n"

    string += "</table>"
    return string