Heatmap: Rotate the graph from horizontal to vertical for easier reading and add the transpose parameter for easier switching in the future

analyze_winscore.py

@@ -219,8 +219,22 @@ def create_heatmap(data_matrix, original_scores,
     y_axis_label="Task", 
     x_axis_visible=True, 
     y_axis_visible=True, 
+    transpose=False, 
 ):
     FONTSIZE = 9
+
+    if transpose:
+        data_matrix = data_matrix.T
+        original_scores = original_scores.T
+        x_axis_label, y_axis_label = y_axis_label, x_axis_label
+        x_axis_visible, y_axis_visible = y_axis_visible, x_axis_visible
+        toolbar_location = "right"
+        x_axis_location = "above"
+        y_range=list(reversed(data_matrix.columns))
+    else:
+        toolbar_location = "below"
+        x_axis_location = "below"
+        y_range=list(data_matrix.columns)
     
     n_rows, n_cols = data_matrix.shape
     cell_size = 22
@@ -245,12 +259,13 @@ def create_heatmap(data_matrix, original_scores,
         width=plot_width,
         height=plot_height,
         x_range=list(data_matrix.index),
-        y_range=list(data_matrix.columns),
-        toolbar_location="below",
+        y_range=y_range,
+        toolbar_location=toolbar_location,
         tools="pan,wheel_zoom,box_zoom,reset,save",
         active_drag=None,
         x_axis_label=x_axis_label,
         y_axis_label=y_axis_label,
+        x_axis_location=x_axis_location,
     )
 
     # Create the color mapper for the heatmap

server.py

@@ -846,12 +846,15 @@ class LeaderboardServer:
         # Apply quantile transformation independently for each row
         normalized_scores_sub = original_scores_sub.apply(lambda x: (x - x.min()) / (x.max() - x.min()), axis=0)
 
+        transpose = True
+
         # Call the heatmap function with the normalized data
         p1 = create_heatmap(
             normalized_scores_sub,
             original_scores_sub * 100,
             x_axis_label="Model ≥16B",
             y_axis_label=fig_y_axis_label,
+            transpose=transpose,
         )
         
         # Smaller models
@@ -867,10 +870,15 @@ class LeaderboardServer:
             x_axis_label="Model <16B",
             y_axis_label=fig_y_axis_label,
             y_axis_visible=False,
+            transpose=transpose,
         )
         
-        from bokeh.layouts import row
-        layout = row(p1, p2)
+        if transpose:
+            from bokeh.layouts import column
+            layout = column(p1, p2)
+        else:
+            from bokeh.layouts import row
+            layout = row(p1, p2)
         
         return layout