update mmlu chart

data_handler.py

@@ -82,13 +82,25 @@ def unified_exam_chart(unified_exam_df, plot_column):
 
 def mmlu_chart(mmlu_df, plot_column):
     df = mmlu_df.copy()
-    subject_cols = ['Biology', 'Business', 'Chemistry', 'Computer Science', 'Economics', 'Engineering', 'Health', 'History', 'Law', 'Math', 'Other', 'Philosophy', 'Physics', 'Psychology']
+
+    subject_cols = [
+        'Biology', 'Business', 'Chemistry', 'Computer Science', 'Economics',
+        'Engineering', 'Health', 'History', 'Law', 'Math', 'Other',
+        'Philosophy', 'Physics', 'Psychology'
+    ]
     df['Average'] = df[subject_cols].mean(axis=1)
-    df = df.sort_values(by=plot_column, ascending=False).reset_index(drop=True)
-    x_col = plot_column
-    title = f'{plot_column}'
-    x_range_max = 1.0
-    fig = px.bar(df,
+
+    df = df.sort_values(by=[plot_column, 'Model'],
+                        ascending=[False, True]
+               ).reset_index(drop=True)
+    
+    x_col        = plot_column
+    title        = f'{plot_column}'
+    x_range_max  = 1.0
+    bar_height_px = 28            
+
+    fig = px.bar(
+        df,
         x=x_col,
         y='Model',
         color=x_col,
@@ -96,15 +108,30 @@ def mmlu_chart(mmlu_df, plot_column):
         labels={x_col: 'Accuracy', 'Model': 'Model'},
         title=title,
         orientation='h',
-        range_color=[0,1]
+        range_color=[0, 1]
     )
 
     fig.update_layout(
+        height=bar_height_px * len(df) + 120,
+        margin=dict(l=220, r=40, t=60, b=40),
+        width=1000,
+
         xaxis=dict(range=[0, x_range_max]),
         title=dict(text=title, font=dict(size=16)),
         xaxis_title=dict(font=dict(size=12)),
         yaxis_title=dict(font=dict(size=12)),
-        yaxis=dict(autorange="reversed"),
-        width=1000
+
+        yaxis=dict(
+            automargin=True,
+            tickmode='array',
+            tickvals=df['Model'],
+            ticktext=df['Model'],
+            dtick=1,
+            autorange='reversed'      
+        )
     )
+
+    fig.update_yaxes(tickfont=dict(size=10))
+
     return fig
+