status-im
/
das-research
mirror of https://github.com/status-im/das-research.git
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Minor display improvements

This commit is contained in:
HajarZaiz 2023-02-19 22:17:50 +01:00
parent c312d6447d
commit 54b0284abb
1 changed files with 14 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
results/plots.py
View File

@ -6,6 +6,12 @@ import numpy as np
import seaborn as sns import seaborn as sns
from itertools import combinations from itertools import combinations
#Title formatting for the figures
def formatTitle(key):
name = ''.join([f" {char}" if char.isupper() else char for char in key.split('_')[0]])
number = key.split('_')[1]
return f"{name.title()}: {number} "
results_folder = os.getcwd() results_folder = os.getcwd()
#Get all folders and store their time info and sort #Get all folders and store their time info and sort
@ -88,24 +94,27 @@ heatmaps_folder = 'heatmaps'
if not os.path.exists(heatmaps_folder): if not os.path.exists(heatmaps_folder):
os.makedirs(heatmaps_folder) os.makedirs(heatmaps_folder)
#Plot the heatmaps and store them in subfolders
for labels, keys in filtered_keys.items(): for labels, keys in filtered_keys.items():
for key in keys: for key in keys:
hist, xedges, yedges = np.histogram2d(data[key][labels[0]], data[key][labels[1]], bins=(3, 3), weights=data[key]['ttas']) hist, xedges, yedges = np.histogram2d(data[key][labels[0]], data[key][labels[1]], bins=(3, 3), weights=data[key]['ttas'])
hist = hist.T hist = hist.T
sns.heatmap(hist, xticklabels=False, yticklabels=False, cmap='Purples', cbar_kws={'label': 'Time to block availability'}, linecolor='black', linewidths=0.3, annot=True, fmt=".2f") xlabels = [f'{val:.2f}' for val in xedges[::2]] + [f'{xedges[-1]:.2f}']
ylabels = [f'{val:.2f}' for val in yedges[::2]] + [f'{yedges[-1]:.2f}']
sns.heatmap(hist, xticklabels=xlabels, yticklabels=ylabels, cmap='Purples', cbar_kws={'label': 'Time to block availability'}, linecolor='black', linewidths=0.3, annot=True, fmt=".2f")
plt.xlabel(labels[0]) plt.xlabel(labels[0])
plt.ylabel(labels[1]) plt.ylabel(labels[1])
filename = ""
title = "" title = ""
paramValueCnt = 0 paramValueCnt = 0
for param in parameters: for param in parameters:
if param != labels[0] and param != labels[1]: if param != labels[0] and param != labels[1]:
title += f"{key[paramValueCnt]}" filename += f"{key[paramValueCnt]}"
title += formatTitle(key[paramValueCnt])
paramValueCnt += 1 paramValueCnt += 1
plt.title(title) plt.title(title)
filename = title + ".png" filename = filename + ".png"
target_folder = os.path.join(heatmaps_folder, f"{labels[0]}Vs{labels[1]}") target_folder = os.path.join(heatmaps_folder, f"{labels[0]}Vs{labels[1]}")
if not os.path.exists(target_folder): if not os.path.exists(target_folder):
os.makedirs(target_folder) os.makedirs(target_folder)
plt.savefig(os.path.join(target_folder, filename)) plt.savefig(os.path.join(target_folder, filename))
plt.clf() plt.clf()