Add few-shot info in graphs

app.py

@@ -319,19 +319,20 @@ def toggle_all_categories(action: str) -> list[gr.CheckboxGroup]:
 def plot_size_vs_score(df: pd.DataFrame, hidden_df: pd.DataFrame) -> go.Figure:
     df2 = hidden_df.iloc[df.index]
     df2 = df2[df2["#Params (B)"] > 0]
-    df2 = df2[["model_name_for_query", "#Params (B)", "AVG"]]
+    df2 = df2[["model_name_for_query", "#Params (B)", "AVG", "Few-shot"]]
     df2["AVG"] = df2["AVG"].astype(float)
-    df2 = df2.rename(columns={"model_name_for_query": "Model"})
+    df2["Few-shot"] = df2["Few-shot"].astype(int)
+    df2 = df2.rename(columns={"model_name_for_query": "Model", "Few-shot": "n-shot"})
     df2["model_name_without_org_name"] = df2["Model"].str.split("/").str[-1]
     fig = px.scatter(
         df2,
         x="#Params (B)",
         y="AVG",
         text="model_name_without_org_name",
-        hover_data=["Model"],
+        hover_data=["Model", "n-shot"],
     )
     fig.update_traces(
-        hovertemplate="<b>%{customdata[0]}</b><br>#Params: %{x:.2f}B<br>AVG: %{y:.4f}<extra></extra>",
+        hovertemplate="<b>%{customdata[0]}</b><br>#Params: %{x:.2f}B<br>n-shot: %{customdata[1]}<br>AVG: %{y:.4f}<extra></extra>",
         textposition="top right",
     )
     fig.update_layout(yaxis_range=[0, 1])
@@ -345,20 +346,21 @@ TASK_AVG_NAME_MAP = {
 
 def plot_average_scores(df: pd.DataFrame, hidden_df: pd.DataFrame) -> go.Figure:
     df2 = hidden_df.iloc[df.index]
-    df2 = df2[["model_name_for_query"] + list(TASK_AVG_NAME_MAP.keys())]
-    df2 = df2.rename(columns={"model_name_for_query": "Model"})
+    df2 = df2[["model_name_for_query", "Few-shot"] + list(TASK_AVG_NAME_MAP.keys())]
+    df2 = df2.rename(columns={"model_name_for_query": "Model", "Few-shot": "n-shot"})
     df2 = df2.rename(columns=TASK_AVG_NAME_MAP)
-    df2 = df2.set_index("Model").astype(float)
+    df2["n-shot"] = df2["n-shot"].astype(int)
+    df2 = df2.set_index(["Model", "n-shot"]).astype(float)
 
     fig = go.Figure()
-    for i, (name, row) in enumerate(df2.iterrows()):
+    for i, ((name, n_shot), row) in enumerate(df2.iterrows()):
         visible = True if i < 3 else "legendonly"  # Display only the first 3 models
         fig.add_trace(
             go.Scatterpolar(
                 r=row.values,
                 theta=row.index,
                 fill="toself",
-                name=name,
+                name=f"{name} ({n_shot}-shot)",
                 hovertemplate="%{theta}: %{r}",
                 visible=visible,
             )