PierreGtch commited on
Commit
76b885d
·
1 Parent(s): 3d0b3bf

Align benchmark display names with paper

Browse files

PhysioNet MI -> PhysioNet-MI, BCIC-2a -> BCIC-IV-2a, ISRUC-SLEEP -> ISRUC.

Files changed (1) hide show
  1. backend/app/config/benchmarks.py +3 -3
backend/app/config/benchmarks.py CHANGED
@@ -28,7 +28,7 @@ EEG_BENCHMARKS: Dict[str, BenchmarkInfo] = {
28
  "bcic2a": BenchmarkInfo(
29
  dataset_id="braindecode/bcic2a",
30
  task_id="motor_imagery_4class",
31
- display_name="BCIC-2a",
32
  description="BCI Competition IV Dataset 2a - 4-class motor imagery classification.",
33
  category="Motor Imagery",
34
  num_classes=4,
@@ -37,7 +37,7 @@ EEG_BENCHMARKS: Dict[str, BenchmarkInfo] = {
37
  "physionet": BenchmarkInfo(
38
  dataset_id="braindecode/physionet",
39
  task_id="motor_imagery_4class",
40
- display_name="PhysioNet MI",
41
  description="PhysioNet Motor Imagery dataset - 4-class motor imagery classification.",
42
  category="Motor Imagery",
43
  num_classes=4,
@@ -55,7 +55,7 @@ EEG_BENCHMARKS: Dict[str, BenchmarkInfo] = {
55
  "isruc_sleep": BenchmarkInfo(
56
  dataset_id="braindecode/isruc-sleep",
57
  task_id="sleep_staging_5class",
58
- display_name="ISRUC-SLEEP",
59
  description="ISRUC-SLEEP dataset - 5-class sleep staging classification.",
60
  category="Sleep Staging",
61
  num_classes=5,
 
28
  "bcic2a": BenchmarkInfo(
29
  dataset_id="braindecode/bcic2a",
30
  task_id="motor_imagery_4class",
31
+ display_name="BCIC-IV-2a",
32
  description="BCI Competition IV Dataset 2a - 4-class motor imagery classification.",
33
  category="Motor Imagery",
34
  num_classes=4,
 
37
  "physionet": BenchmarkInfo(
38
  dataset_id="braindecode/physionet",
39
  task_id="motor_imagery_4class",
40
+ display_name="PhysioNet-MI",
41
  description="PhysioNet Motor Imagery dataset - 4-class motor imagery classification.",
42
  category="Motor Imagery",
43
  num_classes=4,
 
55
  "isruc_sleep": BenchmarkInfo(
56
  dataset_id="braindecode/isruc-sleep",
57
  task_id="sleep_staging_5class",
58
+ display_name="ISRUC",
59
  description="ISRUC-SLEEP dataset - 5-class sleep staging classification.",
60
  category="Sleep Staging",
61
  num_classes=5,