whojavumusic commited on
Commit
e4e1e7d
·
1 Parent(s): 884e11f

Add all moving conditions to the leaderboard

Browse files
Files changed (4) hide show
  1. benchmark/dataset.py +9 -3
  2. init.py +9 -3
  3. metrics_config.py +21 -5
  4. utils_display.py +9 -3
benchmark/dataset.py CHANGED
@@ -17,7 +17,9 @@ PACKED_FILES = {
17
  "low": "packed/low.pt",
18
  "mid": "packed/mid.pt",
19
  "high": "packed/high.pt",
20
- "moving": "packed/moving.pt",
 
 
21
  }
22
 
23
  # Bucket path key -> ``run_evaluation`` result field (``wer_<suffix>``).
@@ -29,7 +31,9 @@ CONDITION_WER_SUFFIX: dict[str, str] = {
29
  "low": "low",
30
  "mid": "mid",
31
  "high": "high",
32
- "moving": "moving",
 
 
33
  }
34
 
35
 
@@ -47,7 +51,9 @@ CONDITION_UI_CHOICES: tuple[tuple[str, str], ...] = (
47
  ("High SNR", "high"),
48
  ("Mid SNR", "mid"),
49
  ("Low SNR", "low"),
50
- ("Moving Sources", "moving"),
 
 
51
  )
52
 
53
  DEFAULT_CONDITION_KEYS: tuple[str, ...] = tuple(PACKED_FILES.keys())
 
17
  "low": "packed/low.pt",
18
  "mid": "packed/mid.pt",
19
  "high": "packed/high.pt",
20
+ "moving_low": "packed/moving_low.pt",
21
+ "moving_mid": "packed/moving_mid.pt",
22
+ "moving_high": "packed/moving_high.pt",
23
  }
24
 
25
  # Bucket path key -> ``run_evaluation`` result field (``wer_<suffix>``).
 
31
  "low": "low",
32
  "mid": "mid",
33
  "high": "high",
34
+ "moving_low": "moving_low",
35
+ "moving_mid": "moving_mid",
36
+ "moving_high": "moving_high",
37
  }
38
 
39
 
 
51
  ("High SNR", "high"),
52
  ("Mid SNR", "mid"),
53
  ("Low SNR", "low"),
54
+ ("Moving Low SNR", "moving_low"),
55
+ ("Moving Mid SNR", "moving_mid"),
56
+ ("Moving High SNR", "moving_high"),
57
  )
58
 
59
  DEFAULT_CONDITION_KEYS: tuple[str, ...] = tuple(PACKED_FILES.keys())
init.py CHANGED
@@ -109,13 +109,17 @@ def leaderboard_row_from_eval_result(
109
  "wer_realistic_high_snr": "wer_high",
110
  "wer_realistic_mid_snr": "wer_mid",
111
  "wer_realistic_low_snr": "wer_low",
112
- "wer_moving_sources": "wer_moving",
 
 
113
  }
114
  _WIRE_KEY_ALIASES: dict[str, tuple[str, ...]] = {
115
  "wer_high": ("wer_high_snr", "wer_sim_high_snr"),
116
  "wer_mid": ("wer_mid_snr",),
117
  "wer_low": ("wer_low_snr", "wer_sim_low_snr"),
118
- "wer_moving": ("wer_moving_sources",),
 
 
119
  }
120
 
121
  def _eval_wer(wire_key: str):
@@ -346,7 +350,9 @@ def load_results() -> pd.DataFrame:
346
  "wer_realistic_high_snr": AutoEvalColumn.wer_realistic_high_snr.name,
347
  "wer_realistic_mid_snr": AutoEvalColumn.wer_realistic_mid_snr.name,
348
  "wer_realistic_low_snr": AutoEvalColumn.wer_realistic_low_snr.name,
349
- "wer_moving_sources": AutoEvalColumn.wer_moving_sources.name,
 
 
350
  }
351
  for csv_key, display_name in scenario_col_map.items():
352
  if csv_key in raw_df.columns:
 
109
  "wer_realistic_high_snr": "wer_high",
110
  "wer_realistic_mid_snr": "wer_mid",
111
  "wer_realistic_low_snr": "wer_low",
112
+ "wer_moving_low": "wer_moving_low",
113
+ "wer_moving_mid": "wer_moving_mid",
114
+ "wer_moving_high": "wer_moving_high",
115
  }
116
  _WIRE_KEY_ALIASES: dict[str, tuple[str, ...]] = {
117
  "wer_high": ("wer_high_snr", "wer_sim_high_snr"),
118
  "wer_mid": ("wer_mid_snr",),
119
  "wer_low": ("wer_low_snr", "wer_sim_low_snr"),
120
+ "wer_moving_low": ("wer_moving_low",),
121
+ "wer_moving_mid": ("wer_moving_mid",),
122
+ "wer_moving_high": ("wer_moving_high",)
123
  }
124
 
125
  def _eval_wer(wire_key: str):
 
350
  "wer_realistic_high_snr": AutoEvalColumn.wer_realistic_high_snr.name,
351
  "wer_realistic_mid_snr": AutoEvalColumn.wer_realistic_mid_snr.name,
352
  "wer_realistic_low_snr": AutoEvalColumn.wer_realistic_low_snr.name,
353
+ "wer_moving_low": AutoEvalColumn.wer_moving_low.name,
354
+ "wer_moving_mid": AutoEvalColumn.wer_moving_mid.name,
355
+ "wer_moving_high": AutoEvalColumn.wer_moving_high.name,
356
  }
357
  for csv_key, display_name in scenario_col_map.items():
358
  if csv_key in raw_df.columns:
metrics_config.py CHANGED
@@ -76,11 +76,27 @@ SCENARIO_METRICS: tuple[ScenarioMetric, ...] = (
76
  status="live",
77
  ),
78
  ScenarioMetric(
79
- key="wer_moving_sources",
80
- label="Moving Sources",
81
- short="Moving Sources",
82
- description="Moving talker / varying geometry (packed moving.pt split).",
83
- group="Realistic conditions",
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
84
  status="live",
85
  ),
86
  )
 
76
  status="live",
77
  ),
78
  ScenarioMetric(
79
+ key="wer_moving_low",
80
+ label="Moving Low SNR",
81
+ short="Moving Low SNR",
82
+ description="Moving talker / varying geometry (packed moving_low.pt split).",
83
+ group="Moving conditions",
84
+ status="live",
85
+ ),
86
+ ScenarioMetric(
87
+ key="wer_moving_mid",
88
+ label="Moving Mid SNR",
89
+ short="Moving Mid SNR",
90
+ description="Moving talker / varying geometry (packed moving_mid.pt split).",
91
+ group="Moving conditions",
92
+ status="live",
93
+ ),
94
+ ScenarioMetric(
95
+ key="wer_moving_high",
96
+ label="Moving High SNR",
97
+ short="Moving High SNR",
98
+ description="Moving talker / varying geometry (packed moving_high.pt split).",
99
+ group="Moving conditions",
100
  status="live",
101
  ),
102
  )
utils_display.py CHANGED
@@ -34,7 +34,9 @@ class AutoEvalColumn:
34
  wer_realistic_high_snr = ColumnContent("High SNR", "number")
35
  wer_realistic_mid_snr = ColumnContent("Mid SNR", "number")
36
  wer_realistic_low_snr = ColumnContent("Low SNR", "number")
37
- wer_moving_sources = ColumnContent("Moving Sources", "number")
 
 
38
  eval_rtf = ColumnContent("RTF", "number")
39
  params_m = ColumnContent("Params", "number")
40
 
@@ -49,7 +51,9 @@ SCENARIO_DISPLAY_COLS: tuple[str, ...] = (
49
  AutoEvalColumn.wer_realistic_high_snr.name,
50
  AutoEvalColumn.wer_realistic_mid_snr.name,
51
  AutoEvalColumn.wer_realistic_low_snr.name,
52
- AutoEvalColumn.wer_moving_sources.name,
 
 
53
  )
54
 
55
  # Leaderboard display header -> canonical CSV / analytics metric key.
@@ -60,7 +64,9 @@ SCENARIO_DISPLAY_TO_KEY: dict[str, str] = {
60
  AutoEvalColumn.wer_realistic_high_snr.name: "wer_realistic_high_snr",
61
  AutoEvalColumn.wer_realistic_mid_snr.name: "wer_realistic_mid_snr",
62
  AutoEvalColumn.wer_realistic_low_snr.name: "wer_realistic_low_snr",
63
- AutoEvalColumn.wer_moving_sources.name: "wer_moving_sources",
 
 
64
  }
65
 
66
 
 
34
  wer_realistic_high_snr = ColumnContent("High SNR", "number")
35
  wer_realistic_mid_snr = ColumnContent("Mid SNR", "number")
36
  wer_realistic_low_snr = ColumnContent("Low SNR", "number")
37
+ wer_moving_low = ColumnContent("Moving Low SNR", "number")
38
+ wer_moving_mid = ColumnContent("Moving Mid SNR", "number")
39
+ wer_moving_high = ColumnContent("Moving High SNR", "number")
40
  eval_rtf = ColumnContent("RTF", "number")
41
  params_m = ColumnContent("Params", "number")
42
 
 
51
  AutoEvalColumn.wer_realistic_high_snr.name,
52
  AutoEvalColumn.wer_realistic_mid_snr.name,
53
  AutoEvalColumn.wer_realistic_low_snr.name,
54
+ AutoEvalColumn.wer_moving_low.name,
55
+ AutoEvalColumn.wer_moving_mid.name,
56
+ AutoEvalColumn.wer_moving_high.name,
57
  )
58
 
59
  # Leaderboard display header -> canonical CSV / analytics metric key.
 
64
  AutoEvalColumn.wer_realistic_high_snr.name: "wer_realistic_high_snr",
65
  AutoEvalColumn.wer_realistic_mid_snr.name: "wer_realistic_mid_snr",
66
  AutoEvalColumn.wer_realistic_low_snr.name: "wer_realistic_low_snr",
67
+ AutoEvalColumn.wer_moving_low.name: "wer_moving_low",
68
+ AutoEvalColumn.wer_moving_mid.name: "wer_moving_mid",
69
+ AutoEvalColumn.wer_moving_high.name: "wer_moving_high",
70
  }
71
 
72