Update stats_data.py

stats_data.py

@@ -14,21 +14,42 @@ SPEAKER_META = {
 }
 
 def get_indomain_breakdown():
+    # Primary Data for Torgo In-Domain (5K Model is the Accuracy Champion)
     data = {
         "Speaker": ["M05", "F01", "M01", "M04", "M02", "M03", "F03", "F04"],
         "Severity": ["Severe", "Severe", "Moderate", "Moderate", "Mild", "Mild", "Mild", "Mild"],
-        "Whisper Tiny": ["12.1%", "12.6%", "32.7%", "31.8%", "62.1%", "58.4%", "61.2%", "59.1%"],
-        "5K Model (Ours)": ["33.1%", "34.2%", "47.2%", "45.6%", "84.5%", "81.8%", "83.5%", "82.8%"],
-        "10K Model (Ours)": ["25.4%", "24.1%", "44.1%", "41.2%", "79.1%", "77.5%", "79.0%", "78.2%"]
+        "Whisper Tiny": [12.1, 12.6, 32.7, 31.8, 62.1, 58.4, 61.2, 59.1],
+        "DSR Lab (5K)": [33.1, 34.2, 47.2, 45.6, 84.5, 81.8, 83.5, 82.8]
     }
-    return pd.DataFrame(data)
+    df = pd.DataFrame(data)
+    
+    # Calculate Gains
+    df["Absolute Gain (%)"] = (df["DSR Lab (5K)"] - df["Whisper Tiny"]).round(2)
+    df["Relative Improvement (%)"] = (((df["DSR Lab (5K)"] - df["Whisper Tiny"]) / df["Whisper Tiny"]) * 100).round(1)
+    
+    # Format for display
+    df["Whisper Tiny"] = df["Whisper Tiny"].astype(str) + "%"
+    df["DSR Lab (5K)"] = df["DSR Lab (5K)"].astype(str) + "%"
+    df["Absolute Gain (%)"] = "+" + df["Absolute Gain (%)"].astype(str) + "%"
+    df["Relative Improvement (%)"] = "+" + df["Relative Improvement (%)"].astype(str) + "%"
+    
+    return df
 
 def get_experimental_summary():
+    # Comparing conditions (In-Domain, LOSO, Zero-Shot)
     data = {
-        "Experiment Condition": ["In-Domain (Seen Torgo)", "LOSO (Unseen Torgo F01)", "Zero-Shot (UA-Speech F02)"],
-        "Whisper Tiny (Baseline)": ["41.50%", "12.38%", "4.33%"],
-        "5K Pure Model Score": ["58.77%", "N/A", "6.19%"],
-        "10K Triple-Mix Score": ["54.67%", "24.76%", "5.98%"],
-        "Relative Gain (Best vs Baseline)": ["+41.6%", "+100.0%", "+42.9%"]
+        "Condition": ["In-Domain (Seen Torgo)", "LOSO (Unseen Torgo F01)", "Zero-Shot (UA-Speech F02)"],
+        "Whisper Tiny": [41.50, 12.38, 4.33],
+        "Our Best Score": [58.77, 24.76, 6.19]
     }
-    return pd.DataFrame(data)
+    df = pd.DataFrame(data)
+    
+    # Calculate Relative Gain
+    df["Relative Gain"] = (((df["Our Best Score"] - df["Whisper Tiny"]) / df["Whisper Tiny"]) * 100).round(1)
+    
+    # Format for display
+    df["Whisper Tiny"] = df["Whisper Tiny"].astype(str) + "%"
+    df["Our Best Score"] = df["Our Best Score"].astype(str) + "%"
+    df["Relative Gain"] = "+" + df["Relative Gain"].astype(str) + "%"
+    
+    return df