1
0
Fork 0
QuaPy/distribution_matching/show_results.py

42 lines
860 B
Python

import sys
from pathlib import Path
import pandas as pd
result_dir = 'results/results_tweet_mae_redohyper'
#result_dir = 'results_lequa_mrae'
dfs = []
pathlist = Path(result_dir).rglob('*.csv')
for path in pathlist:
path_in_str = str(path)
try:
df = pd.read_csv(path_in_str, sep='\t')
df = df[df.iloc[:, 0] != df.columns[0]]
if not df.empty:
dfs.append(df)
except Exception:
print('empty')
df = pd.concat(dfs)
for err in ['MAE', 'MRAE', 'KLD']:
print('-'*100)
print(err)
print('-'*100)
piv = df.pivot_table(index='Dataset', columns='Method', values=err)
piv.loc['mean'] = piv.mean()
pd.set_option('display.max_columns', None)
pd.set_option('display.max_rows', None)
pd.set_option('expand_frame_repr', False)
print(piv)
print()