Corey Morris commited on
Commit
ca8e784
1 Parent(s): c671de9

Added a scatter plot with just the top 50 performing models on MMLU average

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -135,6 +135,12 @@ st.plotly_chart(fig)
135
  fig = create_plot(filtered_data, 'Model Name', 'hellaswag|10', 'MMLU_average')
136
  st.plotly_chart(fig)
137
 
 
 
 
 
 
 
138
  # Add heading to page to say Moral Scenarios
139
  st.header('Moral Scenarios')
140
 
@@ -154,4 +160,5 @@ st.plotly_chart(fig)
154
 
155
  # create a histogram of moral disputes
156
  fig = px.histogram(filtered_data, x="MMLU_moral_disputes", marginal="rug", hover_data=filtered_data.columns)
157
- st.plotly_chart(fig)
 
 
135
  fig = create_plot(filtered_data, 'Model Name', 'hellaswag|10', 'MMLU_average')
136
  st.plotly_chart(fig)
137
 
138
+ # create a new dataframe that only has the 50 highest performing models on MMLU_average
139
+ st.header('Top 50 models on MMLU_average')
140
+ top_50 = filtered_data.nlargest(50, 'MMLU_average')
141
+ fig = create_plot(top_50, 'Model Name', 'arc:challenge|25', 'MMLU_average')
142
+ st.plotly_chart(fig)
143
+
144
  # Add heading to page to say Moral Scenarios
145
  st.header('Moral Scenarios')
146
 
 
160
 
161
  # create a histogram of moral disputes
162
  fig = px.histogram(filtered_data, x="MMLU_moral_disputes", marginal="rug", hover_data=filtered_data.columns)
163
+ st.plotly_chart(fig)
164
+