Spaces:
Running on CPU Upgrade
Running on CPU Upgrade
Commit ·
e4e1e7d
1
Parent(s): 884e11f
Add all moving conditions to the leaderboard
Browse files- benchmark/dataset.py +9 -3
- init.py +9 -3
- metrics_config.py +21 -5
- 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 |
-
"
|
|
|
|
|
|
|
| 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 |
-
"
|
|
|
|
|
|
|
| 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
|
|
|
|
|
|
|
| 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 |
-
"
|
|
|
|
|
|
|
| 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 |
-
"
|
|
|
|
|
|
|
| 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 |
-
"
|
|
|
|
|
|
|
| 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="
|
| 80 |
-
label="Moving
|
| 81 |
-
short="Moving
|
| 82 |
-
description="Moving talker / varying geometry (packed
|
| 83 |
-
group="
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
-
|
|
|
|
|
|
|
| 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.
|
|
|
|
|
|
|
| 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.
|
|
|
|
|
|
|
| 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 |
|