added comparison viz

bio_check/verifier.py

@@ -251,28 +251,37 @@ def visualize_outputs(
 
  return fig
 
- def visualize_comparison(self, data: dict, simulators: list[str], comparison_type: str = 'proximity'):
- # grid plot params
+ def visualize_comparison(self, data: dict, simulators: list, comparison_type='proximity') -> Figure:
  species_data_content = data['content']['results']['results']
  species_names = list(species_data_content.keys())
  num_species = len(species_names)
 
  fig, axes = plt.subplots(nrows=num_species, ncols=3, figsize=(20, 6 * num_species))
+ true_color = '#228B22'
+ false_color = '#DC143C'
 
  if num_species == 1:
  axes = [axes]
 
  for i, species_name in enumerate(species_names):
  for j, simulator_name in enumerate(simulators):
  ax = axes[i][j]
- species_data = data['content']['results']['results'][species_name]
- comparison_data = np.array(list(species_data[comparison_type].values()))
- print(comparison_data.shape)
- # sns.heatmap(ax=ax, data=comparison_data, label=f"Comparison matrix for {species_names}")
- # ax.set_title(f"{species_name} simulation outputs for {simulator_name}")
- # ax.legend()
+ species_data = species_data_content[species_name]
+ comparison_data = [list(col.values()) for col in list(species_data[comparison_type].values())]
+ sns.heatmap(
+ data=comparison_data,
+ ax=ax,
+ xticklabels=simulators,
+ yticklabels=simulators,
+ cmap=[false_color, true_color],
+ linewidths=1,
+ vmin=0,
+ vmax=1
+
+ )
+
+ ax.set_title(f"{species_name} comparison matrix")
 
- # TODO: adjust this
  plt.tight_layout()
  plt.show()