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
|
@@ -7,7 +7,7 @@ tags:
|
|
| 7 |
- dance
|
| 8 |
- p300
|
| 9 |
datasets:
|
| 10 |
-
|
| 11 |
---
|
| 12 |
|
| 13 |
# DANCE — BI2014a P300 event-detection checkpoint
|
|
@@ -68,3 +68,21 @@ model.load_state_dict(
|
|
| 68 |
- Loss: `braindecode.training.DanceLoss` (matched-only IoU normalization).
|
| 69 |
|
| 70 |
Reproduce with `train_checkpoint.py` in this repository.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 7 |
- dance
|
| 8 |
- p300
|
| 9 |
datasets:
|
| 10 |
+
- nm000244
|
| 11 |
---
|
| 12 |
|
| 13 |
# DANCE — BI2014a P300 event-detection checkpoint
|
|
|
|
| 68 |
- Loss: `braindecode.training.DanceLoss` (matched-only IoU normalization).
|
| 69 |
|
| 70 |
Reproduce with `train_checkpoint.py` in this repository.
|
| 71 |
+
|
| 72 |
+
<!-- braindecode:training-data -->
|
| 73 |
+
## Training data
|
| 74 |
+
|
| 75 |
+
The data that shaped these weights. The ids are the `datasets:` field of this card,
|
| 76 |
+
which braindecode reads into `model.card_data.datasets` when you call
|
| 77 |
+
`from_pretrained` (MOABB class names for MOABB datasets, OpenNeuro / NEMAR
|
| 78 |
+
accessions, Hub ids otherwise). `nm`/`on` ids are datasets on [NEMAR](https://nemar.org)
|
| 79 |
+
(`on` = NEMAR's copy of the OpenNeuro dataset with the same number). Rows marked
|
| 80 |
+
`private:` are data the authors did not release; they are not in `datasets:`.
|
| 81 |
+
|
| 82 |
+
| `datasets:` id | Dataset | Stage | Modality |
|
| 83 |
+
|---|---|---|---|
|
| 84 |
+
| [`nm000244`](https://nemar.org/dataset/nm000244) | [Brain Invaders 2014a P300 dataset (MOABB BI2014a)](https://moabb.neurotechx.com/docs/generated/moabb.datasets.BI2014a.html) | training | EEG |
|
| 85 |
+
|
| 86 |
+
- Supervised DANCE event detection, cross-subject. Gradient updates on BI2014a subjects 1, 2, 4, 5, 6, 7, 8, 9 (all their sessions as loaded by MOABBDataset); subject 3 is the held-out subject but its F1-event is used to keep the best epoch (torch.save on improvement), so subject 3 selected the checkpoint. 32 s windows at 128 Hz, 120 epochs per card (script default 100).
|
| 87 |
+
- Sources: https://huggingface.co/braindecode/plot_dance_event_detection/resolve/main/train_checkpoint.py; W/cards-orig/plot_dance_event_detection.md; W/bd/examples/applied_examples/plot_dance_event_detection.py.
|
| 88 |
+
|