├── ADD_MODEL.md ├── DATASETS.md ├── LICENSE ├── README.md ├── dataset_config └── Regression.json ├── engine_for_finetuning.py ├── image ├── few-shot.png ├── overview-of-AdaBrain-Bench.png └── overview.jpeg ├── models ├── EEGConformer.py ├── EEGNet.py ├── EEGPT_mcae.py ├── EEGTransformer.py ├── LMDA.py ├── __init__.py ├── biot.py ├── cbramod.py ├── criss_cross_transformer.py ├── loss.py └── modeling_finetune.py ├── preprocessing ├── BCI-IV-2A │ ├── cross_json_process.py │ ├── data_process.py │ └── multi_json_process.py ├── EEGMAT │ ├── cross_json_process.py │ └── data_process.py ├── HMC │ ├── cross_json_process.py │ └── data_process.py ├── SEED-IV │ ├── cross_json_process.py │ ├── data_process.py │ └── multi_json_process.py ├── SEED-VIG │ ├── cross_json_process.py │ └── data_process.py ├── SEED │ ├── cross_json_process.py │ ├── data_process.py │ └── multi_json_process.py ├── SHHS │ ├── cross_json_process.py │ └── data_process.py ├── SHU │ ├── cross_json_process.py │ ├── data_process.py │ └── multi_json_process.py ├── Siena │ ├── channel_name.py │ ├── cross_json_process.py │ └── data_process.py ├── Sleep-EDF │ ├── cross_json_process.py │ ├── data_process.py │ └── dhedfreader.py ├── TUAB │ ├── cross_json_process.py │ └── data_process.py ├── TUEV │ ├── cross_json_process.py │ └── data_process.py ├── Things-EEG │ ├── data_process.py │ └── multi_json_process.py ├── data_preprocess.sh └── json_process.sh ├── requirements.txt ├── run_finetuning.py └── util ├── eegdatasets.py ├── eegpt_use_channels_names.json ├── optim_factory.py └── utils.py /ADD_MODEL.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/ADD_MODEL.md -------------------------------------------------------------------------------- /DATASETS.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/DATASETS.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/README.md -------------------------------------------------------------------------------- /dataset_config/Regression.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/dataset_config/Regression.json -------------------------------------------------------------------------------- /engine_for_finetuning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/engine_for_finetuning.py -------------------------------------------------------------------------------- /image/few-shot.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/image/few-shot.png -------------------------------------------------------------------------------- /image/overview-of-AdaBrain-Bench.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/image/overview-of-AdaBrain-Bench.png -------------------------------------------------------------------------------- /image/overview.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/image/overview.jpeg -------------------------------------------------------------------------------- /models/EEGConformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/models/EEGConformer.py -------------------------------------------------------------------------------- /models/EEGNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/models/EEGNet.py -------------------------------------------------------------------------------- /models/EEGPT_mcae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/models/EEGPT_mcae.py -------------------------------------------------------------------------------- /models/EEGTransformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/models/EEGTransformer.py -------------------------------------------------------------------------------- /models/LMDA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/models/LMDA.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/biot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/models/biot.py -------------------------------------------------------------------------------- /models/cbramod.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/models/cbramod.py -------------------------------------------------------------------------------- /models/criss_cross_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/models/criss_cross_transformer.py -------------------------------------------------------------------------------- /models/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/models/loss.py -------------------------------------------------------------------------------- /models/modeling_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/models/modeling_finetune.py -------------------------------------------------------------------------------- /preprocessing/BCI-IV-2A/cross_json_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/preprocessing/BCI-IV-2A/cross_json_process.py -------------------------------------------------------------------------------- /preprocessing/BCI-IV-2A/data_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/preprocessing/BCI-IV-2A/data_process.py -------------------------------------------------------------------------------- /preprocessing/BCI-IV-2A/multi_json_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/preprocessing/BCI-IV-2A/multi_json_process.py -------------------------------------------------------------------------------- /preprocessing/EEGMAT/cross_json_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/preprocessing/EEGMAT/cross_json_process.py -------------------------------------------------------------------------------- /preprocessing/EEGMAT/data_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/preprocessing/EEGMAT/data_process.py -------------------------------------------------------------------------------- /preprocessing/HMC/cross_json_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/preprocessing/HMC/cross_json_process.py -------------------------------------------------------------------------------- /preprocessing/HMC/data_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/preprocessing/HMC/data_process.py -------------------------------------------------------------------------------- /preprocessing/SEED-IV/cross_json_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/preprocessing/SEED-IV/cross_json_process.py -------------------------------------------------------------------------------- /preprocessing/SEED-IV/data_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/preprocessing/SEED-IV/data_process.py -------------------------------------------------------------------------------- /preprocessing/SEED-IV/multi_json_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/preprocessing/SEED-IV/multi_json_process.py -------------------------------------------------------------------------------- /preprocessing/SEED-VIG/cross_json_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/preprocessing/SEED-VIG/cross_json_process.py -------------------------------------------------------------------------------- /preprocessing/SEED-VIG/data_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/preprocessing/SEED-VIG/data_process.py -------------------------------------------------------------------------------- /preprocessing/SEED/cross_json_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/preprocessing/SEED/cross_json_process.py -------------------------------------------------------------------------------- /preprocessing/SEED/data_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/preprocessing/SEED/data_process.py -------------------------------------------------------------------------------- /preprocessing/SEED/multi_json_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/preprocessing/SEED/multi_json_process.py -------------------------------------------------------------------------------- /preprocessing/SHHS/cross_json_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/preprocessing/SHHS/cross_json_process.py -------------------------------------------------------------------------------- /preprocessing/SHHS/data_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/preprocessing/SHHS/data_process.py -------------------------------------------------------------------------------- /preprocessing/SHU/cross_json_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/preprocessing/SHU/cross_json_process.py -------------------------------------------------------------------------------- /preprocessing/SHU/data_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/preprocessing/SHU/data_process.py -------------------------------------------------------------------------------- /preprocessing/SHU/multi_json_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/preprocessing/SHU/multi_json_process.py -------------------------------------------------------------------------------- /preprocessing/Siena/channel_name.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/preprocessing/Siena/channel_name.py -------------------------------------------------------------------------------- /preprocessing/Siena/cross_json_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/preprocessing/Siena/cross_json_process.py -------------------------------------------------------------------------------- /preprocessing/Siena/data_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/preprocessing/Siena/data_process.py -------------------------------------------------------------------------------- /preprocessing/Sleep-EDF/cross_json_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/preprocessing/Sleep-EDF/cross_json_process.py -------------------------------------------------------------------------------- /preprocessing/Sleep-EDF/data_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/preprocessing/Sleep-EDF/data_process.py -------------------------------------------------------------------------------- /preprocessing/Sleep-EDF/dhedfreader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/preprocessing/Sleep-EDF/dhedfreader.py -------------------------------------------------------------------------------- /preprocessing/TUAB/cross_json_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/preprocessing/TUAB/cross_json_process.py -------------------------------------------------------------------------------- /preprocessing/TUAB/data_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/preprocessing/TUAB/data_process.py -------------------------------------------------------------------------------- /preprocessing/TUEV/cross_json_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/preprocessing/TUEV/cross_json_process.py -------------------------------------------------------------------------------- /preprocessing/TUEV/data_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/preprocessing/TUEV/data_process.py -------------------------------------------------------------------------------- /preprocessing/Things-EEG/data_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/preprocessing/Things-EEG/data_process.py -------------------------------------------------------------------------------- /preprocessing/Things-EEG/multi_json_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/preprocessing/Things-EEG/multi_json_process.py -------------------------------------------------------------------------------- /preprocessing/data_preprocess.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/preprocessing/data_preprocess.sh -------------------------------------------------------------------------------- /preprocessing/json_process.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/preprocessing/json_process.sh -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_finetuning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/run_finetuning.py -------------------------------------------------------------------------------- /util/eegdatasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/util/eegdatasets.py -------------------------------------------------------------------------------- /util/eegpt_use_channels_names.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/util/eegpt_use_channels_names.json -------------------------------------------------------------------------------- /util/optim_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/util/optim_factory.py -------------------------------------------------------------------------------- /util/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Jamine-W/AdaBrain-Bench/HEAD/util/utils.py --------------------------------------------------------------------------------