Corey Morris commited on
Commit
ac931c6
1 Parent(s): 337b761

added a few charts

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -95,8 +95,8 @@ def create_plot(df, model_column, arc_column, moral_column, models=None):
95
  # Create the scatter plot
96
  fig = px.scatter(plot_data, x=arc_column, y=moral_column, color='color', hover_data=['Model'])
97
  fig.update_layout(showlegend=False, # hide legend
98
- xaxis_title='ARC Accuracy',
99
- yaxis_title='Moral Scenarios Accuracy',
100
  xaxis = dict(),
101
  yaxis = dict())
102
 
@@ -107,3 +107,9 @@ def create_plot(df, model_column, arc_column, moral_column, models=None):
107
 
108
  fig = create_plot(filtered_data, 'Model Name', 'arc:challenge|25', 'moral_scenarios|5')
109
  st.plotly_chart(fig)
 
 
 
 
 
 
 
95
  # Create the scatter plot
96
  fig = px.scatter(plot_data, x=arc_column, y=moral_column, color='color', hover_data=['Model'])
97
  fig.update_layout(showlegend=False, # hide legend
98
+ xaxis_title=arc_column,
99
+ yaxis_title=moral_column,
100
  xaxis = dict(),
101
  yaxis = dict())
102
 
 
107
 
108
  fig = create_plot(filtered_data, 'Model Name', 'arc:challenge|25', 'moral_scenarios|5')
109
  st.plotly_chart(fig)
110
+
111
+ fig = create_plot(filtered_data, 'Model Name', 'arc:challenge|25', 'hellaswag|10')
112
+ st.plotly_chart(fig)
113
+
114
+ fig = create_plot(filtered_data, 'Model Name', 'moral_disputes|5', 'moral_scenarios|5')
115
+ st.plotly_chart(fig)