Subjective ranking of emotions

ressources/emotions_ranked.txt

@@ -0,0 +1,86 @@
+Reachy Mini moves dataset quality ranking
+
+Excellent
+---------
+- anxiety1 (clear)
+- boredom2 (clear)
+- dance2 (clear)
+- dance3 (clear)
+- downcast1 (clear)
+- dying1 (clear)
+- exhausted1 (clear)
+- grateful1 (ambiguous)
+- helpful1 (ambiguous)
+- loving1 (clear)
+- rage1 (clear)
+- reprimand1 (clear)
+- resigned1 (clear)
+- sad1 (clear)
+- sad2 (clear)
+- scared1 (clear)
+- sleep1 (clear)
+- surprised1 (clear)
+- thoughtful1 (clear)
+- welcoming2 (clear)
+
+OK
+--
+- amazed1 (clear)
+- attentive1 (clear)
+- attentive2 (clear)
+- boredom1 (clear)
+- calming1 (ambiguous)
+- come1 (ambiguous)
+- confused1 (clear)
+- contempt1 (ambiguous)
+- curious1 (ambiguous)
+- dance1 (ambiguous)
+- disgusted1 (clear)
+- displeased1 (clear)
+- displeased2 (clear)
+- electric1 (ambiguous)
+- fear1 (clear)
+- frustrated1 (ambiguous)
+- furious1 (ambiguous)
+- go_away1 (ambiguous)
+- helpful2 (ambiguous)
+- impatient1 (ambiguous)
+- impatient2 (clear)
+- incomprehensible2 (ambiguous)
+- indifferent1 (ambiguous)
+- inquiring1 (ambiguous)
+- inquiring2 (ambiguous)
+- inquiring3 (ambiguous)
+- irritated1 (clear)
+- irritated2 (clear)
+- laughing1 (clear)
+- laughing2 (clear)
+- lonely1 (clear)
+- lost1 (ambiguous)
+- no1 (clear)
+- no_excited1 (clear)
+- no_sad1 (clear)
+- proud2 (ambiguous)
+- proud3 (ambiguous)
+- reprimand2 (clear)
+- serenity1 (ambiguous)
+- shy1 (clear)
+- success1 (clear)
+- success2 (clear)
+- surprised2 (clear)
+- thoughtful2 (clear)
+- tired1 (ambiguous)
+- uncertain1 (clear)
+- uncomfortable1 (ambiguous)
+- understanding2 (clear)
+- welcoming1 (ambiguous)
+- yes1 (clear)
+
+Bad
+---
+- cheerful1 (clear)
+- oops1 (ambiguous)
+- oops2 (ambiguous)
+- reprimand3 (clear)
+- understanding1 (clear)
+- yes_sad1 (clear)

ressources/generate_emotions_ranked.py

@@ -0,0 +1,66 @@
+#!/usr/bin/env python3
+"""Generate a simple quality report for the Reachy Mini moves dataset."""
+
+from pathlib import Path
+
+from ruamel.yaml import YAML
+
+
+ROOT = Path(__file__).resolve().parent
+DATASET_PATH = ROOT / "emotions.yml"
+OUTPUT_PATH = ROOT / "emotions_ranked.txt"
+DATASET_NAME = "Reachy Mini moves dataset"
+
+
+def load_moves() -> list[dict]:
+    """Return the list of moves defined in emotions.yml."""
+    yaml = YAML(typ="safe")
+    with DATASET_PATH.open(encoding="utf-8") as stream:
+        data = yaml.load(stream) or {}
+    return data.get("moves", [])
+
+
+def main() -> None:
+    categories: dict[str, list[tuple[str, str]]] = {
+        "excellent": [],
+        "ok": [],
+        "bad": [],
+    }
+
+    for move in load_moves():
+        quality = (move.get("quality") or "").strip().lower()
+        if quality not in categories:
+            continue
+
+        name = move.get("id") or move.get("clean_name") or "unnamed_move"
+        precision = (move.get("precision") or "ambiguous").strip().lower()
+        precision_label = "clear" if precision == "clear" else "ambiguous"
+        categories[quality].append((name, precision_label))
+
+    sections: list[str] = [
+        f"{DATASET_NAME} quality ranking",
+        "",
+    ]
+
+    quality_labels = [
+        ("excellent", "Excellent"),
+        ("ok", "OK"),
+        ("bad", "Bad"),
+    ]
+
+    for key, label in quality_labels:
+        sections.append(label)
+        sections.append("-" * len(label))
+        entries = sorted(categories[key], key=lambda item: item[0].lower())
+        if not entries:
+            sections.append("- None")
+        else:
+            for name, precision in entries:
+                sections.append(f"- {name} ({precision})")
+        sections.append("")
+
+    OUTPUT_PATH.write_text("\n".join(sections).strip() + "\n", encoding="utf-8")
+
+
+if __name__ == "__main__":
+    main()