IliaLarchenko commited on
Commit
9bc730f
1 Parent(s): 7c9c7c8

Added response time and number of messages to the report

Browse files
Files changed (1) hide show
  1. tests/analysis.py +2 -2
tests/analysis.py CHANGED
@@ -89,9 +89,9 @@ def generate_and_display_tables(df):
89
  grouped_by_type_styled = grouped_by_type.style.map(highlight_color)
90
  grouped_by_type_styled.set_caption("Scores Grouped by Unique Type")
91
 
92
- # Grouped by unique interviewer model and sorted by descending total score
93
  total_llm_scores = df.groupby("agent_llm")[prefix_columns].mean().mean(axis=1).sort_values(ascending=False)
94
- grouped_by_interviewer = pd.DataFrame(total_llm_scores, columns=["avg score"])
 
95
  grouped_by_interviewer_styled = grouped_by_interviewer.style.map(highlight_color)
96
  grouped_by_interviewer_styled.set_caption("Scores Grouped by Unique Interviewer Model")
97
 
 
89
  grouped_by_type_styled = grouped_by_type.style.map(highlight_color)
90
  grouped_by_type_styled.set_caption("Scores Grouped by Unique Type")
91
 
 
92
  total_llm_scores = df.groupby("agent_llm")[prefix_columns].mean().mean(axis=1).sort_values(ascending=False)
93
+ # Grouped by unique interviewer model and sorted by descending total score
94
+ grouped_by_interviewer = df.groupby("agent_llm")[["overall_score", "average_response_time_seconds", "number_of_messages"]].mean()
95
  grouped_by_interviewer_styled = grouped_by_interviewer.style.map(highlight_color)
96
  grouped_by_interviewer_styled.set_caption("Scores Grouped by Unique Interviewer Model")
97