From 5a38002db388edbfe34d8b8f2a8b90df1ece8823 Mon Sep 17 00:00:00 2001 From: Max Berrendorf Date: Tue, 16 Jun 2020 16:34:28 +0200 Subject: [PATCH] Adjust generate size table --- reproducibility/generate_summary_table.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/reproducibility/generate_summary_table.py b/reproducibility/generate_summary_table.py index d5f6a4cc1..2497bbb4b 100644 --- a/reproducibility/generate_summary_table.py +++ b/reproducibility/generate_summary_table.py @@ -219,10 +219,8 @@ def generate_size_table(): rv = df[['dataset', 'model', 'model_bytes']].drop_duplicates() rv['dataset'] = rv['dataset'].map(lambda s: pykeen.datasets.datasets[s].__name__) rv['Bytes'] = rv['model_bytes'].map(humanize.naturalsize) - rv['Parameters'] = rv['model_bytes'].map(lambda s: humanize.naturalsize(int(s) / 4).rstrip('B')) - rv.rename(columns={'dataset': 'Dataset', 'model': 'Model'}, inplace=True) - del rv['model_bytes'] - rv = rv.sort_values(['Dataset', 'Model']).set_index(['Dataset', 'Model']) + rv = rv.pivot(index='model', columns='dataset', values='Bytes').fillna('-') + rv = rv[sorted(rv.columns)] rv.to_csv(os.path.join(SUMMARIES, 'sizes.tsv'), sep='\t') with open(os.path.join(SUMMARIES, 'sizes.tex'), 'w') as file: print(rv.to_latex(multirow=True), file=file)