graphs.py

#!/usr/bin/env python3

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.loc[df_2['lib'] == 'Pinocchio'].groupby('algo').boxplot(by=['model'], sym='')
    # df_2.loc[df_2['model'] == 'lwr']].groupby('algo').boxplot(by=['lib', 'model'], sym='')