Card: list the training data (datasets:, base_model:) and add a Training data section
Browse filesFollow-up to braindecode PR #1493: `from_pretrained` reads `datasets:` into `model.card_data`. Ids: MOABB names, NEMAR nm/on ids, OpenNeuro ds ids; evidence per checkpoint.
README.md
CHANGED
|
@@ -2,6 +2,25 @@
|
|
| 2 |
license: mit
|
| 3 |
tags:
|
| 4 |
- braindecode
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 5 |
---
|
| 6 |
|
| 7 |
Shallow conversion from the original weight for braindecode.
|
|
@@ -303,4 +322,41 @@ Examples:
|
|
| 303 |
print(f"Skipped: {stats['skipped']}")
|
| 304 |
print("="*70)
|
| 305 |
|
| 306 |
-
```
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
license: mit
|
| 3 |
tags:
|
| 4 |
- braindecode
|
| 5 |
+
datasets:
|
| 6 |
+
- BCIC-IV-1
|
| 7 |
+
- Emobrain
|
| 8 |
+
- WAY-EEG-GAL
|
| 9 |
+
- Inria-BCI-Challenge
|
| 10 |
+
- on004362
|
| 11 |
+
- Trujillo2020
|
| 12 |
+
- Trujillo2017
|
| 13 |
+
- SEED
|
| 14 |
+
- SEED-IV
|
| 15 |
+
- SEED-GER
|
| 16 |
+
- SEED-FRA
|
| 17 |
+
- nm000385
|
| 18 |
+
- SPIS-Resting-State
|
| 19 |
+
- nm000216
|
| 20 |
+
- TUAR
|
| 21 |
+
- TUEP
|
| 22 |
+
- TUSZ
|
| 23 |
+
- TUSL
|
| 24 |
---
|
| 25 |
|
| 26 |
Shallow conversion from the original weight for braindecode.
|
|
|
|
| 322 |
print(f"Skipped: {stats['skipped']}")
|
| 323 |
print("="*70)
|
| 324 |
|
| 325 |
+
```
|
| 326 |
+
|
| 327 |
+
<!-- braindecode:training-data -->
|
| 328 |
+
## Training data
|
| 329 |
+
|
| 330 |
+
The data that shaped these weights. The ids are the `datasets:` field of this card,
|
| 331 |
+
which braindecode reads into `model.card_data.datasets` when you call
|
| 332 |
+
`from_pretrained` (MOABB class names for MOABB datasets, OpenNeuro / NEMAR
|
| 333 |
+
accessions, Hub ids otherwise). `nm`/`on` ids are datasets on [NEMAR](https://nemar.org)
|
| 334 |
+
(`on` = NEMAR's copy of the OpenNeuro dataset with the same number). Rows marked
|
| 335 |
+
`private:` are data the authors did not release; they are not in `datasets:`.
|
| 336 |
+
|
| 337 |
+
| `datasets:` id | Dataset | Stage | Modality |
|
| 338 |
+
|---|---|---|---|
|
| 339 |
+
| `BCIC-IV-1` | [BCI Competition IV dataset 1 (Blankertz et al. 2007; 59 ch, 7 subjects, motor imagery)](https://www.bbci.de/competition/iv/desc_1.html) | pretraining | EEG |
|
| 340 |
+
| `Emobrain` | Emobrain, eNTERFACE'06 multimodal emotion dataset (Savran et al. 2006; 64-ch EEG + fNIRS, 16 subjects) | pretraining | EEG |
|
| 341 |
+
| `WAY-EEG-GAL` | [Grasp-and-Lift EEG Challenge (Luciw et al. 2014; 32 ch, 12 subjects)](https://www.kaggle.com/c/grasp-and-lift-eeg-detection) | pretraining | EEG |
|
| 342 |
+
| `Inria-BCI-Challenge` | [Inria BCI Challenge / Kaggle ERN (Margaux et al. 2012; P300 speller error-related potentials, 26 subjects)](https://www.kaggle.com/c/inria-bci-challenge) | pretraining | EEG |
|
| 343 |
+
| [`on004362`](https://nemar.org/dataset/on004362) | [EEG Motor Movement/Imagery Dataset (EEGMMIDB, Schalk et al. 2004; 109 subjects)](https://physionet.org/content/eegmmidb/1.0.0/) | pretraining | EEG |
|
| 344 |
+
| `Trujillo2020` | [Raw EEG Data (Trujillo 2020; information-integration and rule-based categorization tasks, 64 ch)](https://doi.org/10.18738/T8/SS2NHB) | pretraining | EEG |
|
| 345 |
+
| `Trujillo2017` | [Resting State EEG Data (Trujillo et al. 2017; 22 subjects, 64 ch)](https://doi.org/10.3389/fnins.2017.00425) | pretraining | EEG |
|
| 346 |
+
| `SEED` | [SJTU Emotion EEG Dataset (15 subjects)](https://bcmi.sjtu.edu.cn/home/seed/) | pretraining | EEG |
|
| 347 |
+
| `SEED-IV` | [SJTU Emotion EEG Dataset IV (15 subjects)](https://bcmi.sjtu.edu.cn/home/seed/) | pretraining | EEG |
|
| 348 |
+
| `SEED-GER` | [SEED-GER, German-subject emotion EEG (8 subjects; Liu et al. 2022)](https://bcmi.sjtu.edu.cn/home/seed/) | pretraining | EEG |
|
| 349 |
+
| `SEED-FRA` | [SEED-FRA, French-subject emotion EEG (8 subjects; Liu et al. 2022)](https://bcmi.sjtu.edu.cn/home/seed/) | pretraining | EEG |
|
| 350 |
+
| [`nm000385`](https://nemar.org/dataset/nm000385) | [Siena Scalp EEG Database (Detti et al. 2020; 14 patients)](https://physionet.org/content/siena-scalp-eeg/1.0.0/) | pretraining | EEG |
|
| 351 |
+
| `SPIS-Resting-State` | [SPIS Resting State Dataset (Torkamani-Azar et al. 2020; 10 subjects)](https://github.com/mastaneht/SPIS-Resting-State-Dataset) | pretraining | EEG |
|
| 352 |
+
| [`nm000216`](https://nemar.org/dataset/nm000216) | [Brain Invaders calibration-less P300 BCI with modulation of flash duration (bi2015a; Korczowski et al. 2019) - LaBraM's 'Target Versus Non-Target'](https://zenodo.org/record/3266930) | pretraining | EEG |
|
| 353 |
+
| `TUAR` | [TUH EEG Artifact Corpus](https://isip.piconepress.com/projects/nedc/html/tuh_eeg/) | pretraining | EEG |
|
| 354 |
+
| `TUEP` | [TUH EEG Epilepsy Corpus](https://isip.piconepress.com/projects/nedc/html/tuh_eeg/) | pretraining | EEG |
|
| 355 |
+
| `TUSZ` | [TUH EEG Seizure Corpus](https://isip.piconepress.com/projects/nedc/html/tuh_eeg/) | pretraining | EEG |
|
| 356 |
+
| `TUSL` | [TUH EEG Slowing Corpus](https://isip.piconepress.com/projects/nedc/html/tuh_eeg/) | pretraining | EEG |
|
| 357 |
+
| `private:LaBraM-self-collected-EEG` | LaBraM authors' self-collected EEG (>140 subjects, 62 ch, 1000 Hz, ESI NeuroScan; 342.23 h; not released) | pretraining | EEG |
|
| 358 |
+
|
| 359 |
+
- same corpus for the vector-quantized neural tokenizer (vqnsp) and masked-code pretraining; 2,534.78 h total; TUAB, TUEV, SEED-V and MoBI (downstream) excluded per paper
|
| 360 |
+
- **Incomplete list.** Same as braindecode/labram-pretrained: private self-collected EEG not itemised; released-checkpoint cohort unconfirmed. A dataset missing here is therefore not proof that these weights never saw it.
|
| 361 |
+
- Sources: arXiv:2405.18765 Appendix D (Pre-training Dataset Description); arXiv:2405.18765 Appendix D; arXiv:2405.18765 References; arXiv:2405.18765 Sec. 3 Experiments, paragraph 'Pre-training & Fine-tuning'.
|
| 362 |
+
|