#!/usr/bin/env python

from pathlib import Path

import pandas


PATH = Path('data')

data = {}
for logs in PATH.glob('*.txt'):
    with logs.open() as f:
        data[logs.stem] = [int(i) for i in f.readlines()]

df = pandas.DataFrame(data)

datasets = [f.stem.split('_') for f in PATH.glob('*.txt')]
hostnames, libs, algos, models = (set(d[i] for d in datasets) for i in range(4))


if __name__ == '__main__':
    data_2 = []
    for log in PATH.glob('*.txt'):
        with log.open() as f:
            for line in f.readlines():
                data_2.append(log.stem.split('_') + [int(line)])
    df_2 = pandas.DataFrame(data_2, columns=['hostname', 'lib', 'algo', 'model', 'time'])
    df_2.groupby('model').boxplot(by=['lib', 'algo'], sym='')