Heatmap: Enabling visibility of the Y axis as an attribute and setting the font using a constant

analyze_winscore.py

@@ -215,7 +215,20 @@ def create_heatmap(data_matrix, original_scores,
     plot_height=None, 
     x_axis_label="Model", 
     y_axis_label="Task", 
+    x_axis_visible=True, 
+    y_axis_visible=True, 
 ):
+    FONTSIZE = 13
+    
+    n_rows, n_cols = data_matrix.shape
+    y_axis_size = 0
+    x_axis_size = 0
+    cell_width_border = 1 + 3
+    cell_height_border = 4 + 3
+    cell_size = 30
+    plot_width  = n_rows * cell_size + (n_rows * cell_width_border + y_axis_size if y_axis_visible else 0)
+    plot_height = n_cols * cell_size + (n_cols * cell_height_border + x_axis_size if x_axis_visible else 0)
+
     if selected_rows is not None:
         # Select only the specified rows (models)
         data_matrix = data_matrix[selected_rows]
@@ -223,6 +236,7 @@ def create_heatmap(data_matrix, original_scores,
 
     # Set up the figure with tasks as x-axis and models as y-axis
     p = figure(
+        sizing_mode="fixed",
         width=plot_width,
         height=plot_height, 
         x_range=list(data_matrix.index),
@@ -297,16 +311,18 @@ def create_heatmap(data_matrix, original_scores,
     p.xgrid.grid_line_color = None
     p.ygrid.grid_line_color = None
     p.xaxis.major_label_orientation = "vertical"
-    p.yaxis.major_label_text_font_size = "13pt"
-    p.xaxis.major_label_text_font_size = "13pt"
+    p.yaxis.major_label_text_font_size = f"{FONTSIZE}pt"
+    p.xaxis.major_label_text_font_size = f"{FONTSIZE}pt"
 
     # Set the axis label font size
-    p.xaxis.axis_label_text_font_size = "18pt"  # Set font size for x-axis label
-    p.yaxis.axis_label_text_font_size = "18pt"  # Set font size for y-axis label
+    p.xaxis.axis_label_text_font_size = f"{FONTSIZE + 5}pt"  # Set font size for x-axis label
+    p.yaxis.axis_label_text_font_size = f"{FONTSIZE + 5}pt"  # Set font size for y-axis label
     p.xaxis.axis_label_text_font_style = "normal"  # Set x-axis label to normal
     p.yaxis.axis_label_text_font_style = "normal"  # Set y-axis label to normal
     
-    #p.yaxis.visible = False  # Hide the y-axis labels
+    # Hide the axis labels
+    p.xaxis.visible = x_axis_visible
+    p.yaxis.visible = y_axis_visible
 
     return p