alex/copycat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Further cleans resulting tables

Browse Source
This commit is contained in:
LSaldyt
2018-01-12 15:09:13 -07:00
parent 6215ccb08a
commit 1506e673a2
1 changed files with 13 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
cross_compare.py
View File

@ -3,6 +3,7 @@ import sys
import pickle import pickle
from pprint import pprint from pprint import pprint
from collections import defaultdict
from copycat import Problem from copycat import Problem
from copycat.statistics import cross_table from copycat.statistics import cross_table
@ -26,22 +27,23 @@ def main(args):
headKey, headSubDict = tableItems[0] headKey, headSubDict = tableItems[0]
# Create table and add headers # Create table and add headers
table = [['problems x variants']] table = [['source', 'compare', 'source formula', 'compare formula']]
for subkey, subsubdict in key_sorted_items(headSubDict): for key, _ in tableItems:
for subsubkey, result in key_sorted_items(subsubdict):
table[-1].append('{} x {} for {} x {}'.format(*subkey, *subsubkey))
# Add test results to table
for key, subdict in tableItems:
table.append([])
problem = '{}:{}::{}:_'.format(*key) problem = '{}:{}::{}:_'.format(*key)
table[-1].append(problem) table[-1].append(problem)
# Arranged results in terms of copycat variants and formulas
arranged = defaultdict(list)
for key, subdict in tableItems:
for subkey, subsubdict in key_sorted_items(subdict): for subkey, subsubdict in key_sorted_items(subdict):
for subsubkey, result in key_sorted_items(subsubdict): for subsubkey, result in key_sorted_items(subsubdict):
table[-1].append(str(result)) arranged[subkey + subsubkey].append((key, result))
# A simple transpose # Add test results to table
table = list(map(list, zip(*table))) for key, results in arranged.items():
table.append(list(map(str, [*key])))
for _, result in results:
table[-1].append(str(result))
with open('output/cross_compare.csv', 'w') as outfile: with open('output/cross_compare.csv', 'w') as outfile:
outfile.write('\n'.join(','.join(row) for row in table) + '\n') outfile.write('\n'.join(','.join(row) for row in table) + '\n')