[CS598 DLH] Add DREAMT OSA classification #709
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Title
Add DREAMT dataset + OSA classification task
Summary
This PR adds support for the DREAMT sleep dataset and a simple patient-level OSA classification task based on DREAMT clinical metadata.
Changes
DREAMTDataset
BaseDataset.dreamt-metadata.csvfromparticipant_info.csv.SID → patient_id,AHI → ahi, etc.) and attaches paths to 64Hz/100Hz CSVs.DREAMTOSAClassification
BaseTaskfor patient-level OSA prediction.ahi_severity_4class(normal / mild / moderate / severe).ahi_binary_15(AHI < 15 vs ≥ 15).oahi_binary_5(OAHI < 5 vs ≥ 5).Tests
tests/core/test_dreamt_osa_task.py:ahi_severity_4class.ahi_binary_15.