Skip to content

EEG Model PR: BIOT#849

Open
Jathurshan0330 wants to merge 3 commits intomasterfrom
biot
Open

EEG Model PR: BIOT#849
Jathurshan0330 wants to merge 3 commits intomasterfrom
biot

Conversation

@Jathurshan0330
Copy link
Collaborator

Adds BIOT model (Yang et al., NeurIPS 2023) for biosignal (EEG) classification to pyhealth. models.

Files to check:

  • New pyhealth/models/biot.py with BIOT, BIOTEncoder, BIOTClassifier
  • pyhealth/models/init.py
  • tests/core/test_biot.py
  • example ./eeg/eeg_models/BIOT_tuev_eeg_event_classification.ipynb

Also renamed the folders to eeg_models under examples: this is were the EEG model examples will be added.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant