Update README.md

README.md

@@ -94,15 +94,47 @@ MLP_DROPOUTS = [0.0, 0.1, 0.1, 0.1] # For 'Small' models
 
 # Mapping from .pth file name parts to human-readable dimension keys
 # (Abridged, full map in Colab Cell 2)
-FILENAME_PART_TO_TARGET_KEY_MAP = {
-    "Affection": "Affection", "Amusement": "Amusement", "Anger": "Anger",
-    "Arousal": "Arousal", "Valence": "Valence", # ... and many more
-    # Add all 40 emotions and other attributes as per Colab Cell 2
+
+FILENAME_PART_TO_TARGET_KEY_MAP: Dict[str, str] = {
+    "Affection": "Affection", "Age": "Age", "Amusement": "Amusement", "Anger": "Anger",
+    "Arousal": "Arousal", "Astonishment_Surprise": "Astonishment/Surprise",
+    "Authenticity": "Authenticity", "Awe": "Awe", "Background_Noise": "Background_Noise",
+    "Bitterness": "Bitterness", "Concentration": "Concentration",
+    "Confident_vs._Hesitant": "Confident_vs._Hesitant", "Confusion": "Confusion",
+    "Contemplation": "Contemplation", "Contempt": "Contempt", "Contentment": "Contentment",
+    "Disappointment": "Disappointment", "Disgust": "Disgust", "Distress": "Distress",
+    "Doubt": "Doubt", "Elation": "Elation", "Embarrassment": "Embarrassment",
+    "Emotional_Numbness": "Emotional Numbness", "Fatigue_Exhaustion": "Fatigue/Exhaustion",
+    "Fear": "Fear", "Gender": "Gender", "Helplessness": "Helplessness",
+    "High-Pitched_vs._Low-Pitched": "High-Pitched_vs._Low-Pitched",
+    "Hope_Enthusiasm_Optimism": "Hope/Enthusiasm/Optimism",
+    "Impatience_and_Irritability": "Impatience and Irritability",
+    "Infatuation": "Infatuation", "Interest": "Interest",
+    "Intoxication_Altered_States_of_Consciousness": "Intoxication/Altered States of Consciousness",
+    "Jealousy_&_Envy": "Jealousy / Envy", "Longing": "Longing",
+    "Malevolence_Malice": "Malevolence/Malice",
+    "Monotone_vs._Expressive": "Monotone_vs._Expressive", "Pain": "Pain",
+    "Pleasure_Ecstasy": "Pleasure/Ecstasy", "Pride": "Pride",
+    "Recording_Quality": "Recording_Quality", "Relief": "Relief", "Sadness": "Sadness",
+    "Serious_vs._Humorous": "Serious_vs._Humorous", "Sexual_Lust": "Sexual Lust",
+    "Shame": "Shame", "Soft_vs._Harsh": "Soft_vs._Harsh", "Sourness": "Sourness",
+    "Submissive_vs._Dominant": "Submissive_vs._Dominant", "Teasing": "Teasing",
+    "Thankfulness_Gratitude": "Thankfulness/Gratitude", "Triumph": "Triumph",
+    "Valence": "Valence",
+    "Vulnerable_vs._Emotionally_Detached": "Vulnerable_vs._Emotionally_Detached",
+    "Warm_vs._Cold": "Warm_vs._Cold"
 }
-TARGET_EMOTION_KEYS_FOR_REPORT = [ # The 40 primary emotions
-    "Amusement", "Elation", # ... (full list from Colab Cell 2)
-]
 
+TARGET_EMOTION_KEYS_FOR_REPORT: List[str] = [
+    "Amusement", "Elation", "Pleasure/Ecstasy", "Contentment", "Thankfulness/Gratitude",
+    "Affection", "Infatuation", "Hope/Enthusiasm/Optimism", "Triumph", "Pride",
+    "Interest", "Awe", "Astonishment/Surprise", "Concentration", "Contemplation",
+    "Relief", "Longing", "Teasing", "Impatience and Irritability",
+    "Sexual Lust", "Doubt", "Fear", "Distress", "Confusion", "Embarrassment", "Shame",
+    "Disappointment", "Sadness", "Bitterness", "Contempt", "Disgust", "Anger",
+    "Malevolence/Malice", "Sourness", "Pain", "Helplessness", "Fatigue/Exhaustion",
+    "Emotional Numbness", "Intoxication/Altered States of Consciousness", "Jealousy / Envy"
+]
 
 # --- MLP Model Definition (from Colab Cell 2) ---
 class FullEmbeddingMLP(nn.Module):