Update my_model/results/demo.py

my_model/results/demo.py

@@ -96,7 +96,7 @@ class ResultDemonstrator:
 
     def plot_token_count_vs_scores(self, conf: str, model_name: str, score_name: str = 'VQA Score') -> None:
         """
-        Plots an interactive scatter plot comparing token counts to VQA or EM scores using Altair.
+        Plots an interactive scatter plot comparing token count to VQA or EM scores using Altair.
 
         Args:
             conf (str): The configuration name.
@@ -125,13 +125,13 @@ class ResultDemonstrator:
             legend_map = ['Correct' if score == 1 else 'Incorrect' for score in scores]
             color_scale = alt.Scale(domain=['Correct', 'Incorrect'], range=['green', 'red'])
 
-        # Retrieve token counts from the data
-        token_counts = self.main_data[f'tokens_count_{conf}']
+        # Retrieve token count from the data
+        token_count = self.main_data[f'tokens_count_{conf}']
 
         # Create a DataFrame for the scatter plot
         scatter_data = pd.DataFrame({
-            'Index': range(len(token_counts)),
-            'Token Count': token_counts,
+            'Index': range(len(token_count)),
+            'Token Count': token_count,
             score_name: legend_map
         })
 
@@ -143,14 +143,14 @@ class ResultDemonstrator:
             stroke='black'  # Sets the border color to black
         ).encode(
             x=alt.X('Index', scale=alt.Scale(domain=[0, 1020])),
-            y=alt.Y('Token Counts', scale=alt.Scale(domain=[token_counts.min()-200, token_counts.max()+200])),
+            y=alt.Y('Token Count', scale=alt.Scale(domain=[token_count.min()-200, token_count.max()+200])),
             color=alt.Color(score_name, scale=color_scale, legend=alt.Legend(title=score_name)),
-            tooltip=['Index', 'Token Counts', score_name]
+            tooltip=['Index', 'Token Count', score_name]
         ).interactive()  # Enables zoom & pan
 
         chart = chart.properties(
             title={
-                "text": f"Token Counts vs {score_name} ({model_configuration})",
+                "text": f"Token Count vs {score_name} ({model_configuration})",
                 "color": "black",  # Optional color
                 "fontSize": 20,  # Optional font size
                 "anchor": "middle",  # Optional anchor position