alfraser commited on
Commit
dc63ce8
1 Parent(s): 1fb12dc

Added a time/response length plot to the charts

Browse files
Files changed (1) hide show
  1. pages/040_Test_Reporter.py +10 -0
pages/040_Test_Reporter.py CHANGED
@@ -44,6 +44,16 @@ def show_stats(for_test_group: str):
44
  fig.update_xaxes(tickangle=-90)
45
  st.plotly_chart(fig, use_container_width=True)
46
 
 
 
 
 
 
 
 
 
 
 
47
 
48
  if st_setup('LLM Arch'):
49
  summary = st.container()
 
44
  fig.update_xaxes(tickangle=-90)
45
  st.plotly_chart(fig, use_container_width=True)
46
 
47
+ with st.expander("**Time by reponse length (seconds/char)**"):
48
+ data = []
49
+ for arch in stats:
50
+ for elapsed, resp_len in zip(arch['elapsed'], arch['response_len']):
51
+ data.append([arch['arch_name'], elapsed / 1000, resp_len])
52
+ df = pd.DataFrame(data, columns=['Architecture', 'Elapsed time', 'Response length'])
53
+ fig = px.scatter(df, x='Elapsed time', y='Response length', color='Architecture')
54
+ fig.update_xaxes(tickangle=-90)
55
+ st.plotly_chart(fig, use_container_width=True)
56
+
57
 
58
  if st_setup('LLM Arch'):
59
  summary = st.container()