Bagratuni commited on
Commit
119f7c7
·
1 Parent(s): 2344677
Files changed (1) hide show
  1. data_handler.py +14 -1
data_handler.py CHANGED
@@ -56,14 +56,27 @@ def unified_exam_chart(unified_exam_df, plot_column):
56
  orientation='h'
57
  )
58
 
 
 
59
  fig.update_layout(
 
 
60
  xaxis=dict(range=[0, x_range_max]),
61
  title=dict(text=title, font=dict(size=16)),
62
  xaxis_title=dict(font=dict(size=12)),
63
  yaxis_title=dict(font=dict(size=12)),
64
- yaxis=dict(autorange="reversed"),
 
 
 
 
 
 
65
  width=1000
66
  )
 
 
 
67
  return fig
68
 
69
  def mmlu_chart(mmlu_df, plot_column):
 
56
  orientation='h'
57
  )
58
 
59
+ bar_height_px = 28
60
+
61
  fig.update_layout(
62
+ height=bar_height_px * len(df) + 120,
63
+ margin=dict(l=220, r=40, t=60, b=40),
64
  xaxis=dict(range=[0, x_range_max]),
65
  title=dict(text=title, font=dict(size=16)),
66
  xaxis_title=dict(font=dict(size=12)),
67
  yaxis_title=dict(font=dict(size=12)),
68
+ yaxis=dict(
69
+ automargin=True,
70
+ tickmode="array",
71
+ tickvals=df["Model"],
72
+ ticktext=df["Model"],
73
+ dtick=1
74
+ ),
75
  width=1000
76
  )
77
+
78
+ fig.update_yaxes(tickfont=dict(size=10))
79
+
80
  return fig
81
 
82
  def mmlu_chart(mmlu_df, plot_column):