Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Activity count feature addition #249

Merged
merged 43 commits into from
Feb 13, 2025

Conversation

simon-p-2000
Copy link
Contributor

Added the option to download, process and read the activity counts from the mesa dataset. This is done by extending the read_mesa function. The activity counts represent processed accelerometer data and can be used as an additional feature to train the existing classifiers.

The option to download and process the activity counts is integrated into the read_mesa function by a variable activity_source and is set to False by default. As such, unless specified, no activity counts will be used. If either cached activity counts or the actigraphy csv files in the mesa database are used, the SleepRecord which the function returns, contains an additional array with the activity counts for the specified recording.

This new feature is integrated in such a way that it works for both the offline and non offline option provided by the read_mesa function. Since the actigraphy csv files are much smaller than the EDF files, no option to delete the csv files afer downloading them is provided.

tests/test_sleep_readers.py Outdated Show resolved Hide resolved
@cbrnr
Copy link
Owner

cbrnr commented Jan 14, 2025

I have made two cosmetic commits (please check if you're OK with the changes).

tests/test_sleep_readers.py Outdated Show resolved Hide resolved
@simon-p-2000
Copy link
Contributor Author

grafik

Running the ws_gru_mesa classifier without the activity feature leads to this result

@cbrnr
Copy link
Owner

cbrnr commented Feb 11, 2025

Yep, this looks good! Can you run the other two as well and compare with https://sleepecg.readthedocs.io/en/stable/classification/?

@simon-p-2000
Copy link
Contributor Author

wrn_gru_mesa_weighted:
grafik

@simon-p-2000
Copy link
Contributor Author

grafik
wrn_gru_mesa classifier

It appears that the classifiers are working similarly to before the addition of the activity count feature, so when the activity counts are not added as a feature the classifier is not affected. The next step would be to run the classifiers again with the activity counts added as a feature and see if that improves the classifier performance.

docs/datasets.md Outdated Show resolved Hide resolved
CHANGELOG.md Outdated Show resolved Hide resolved
@cbrnr cbrnr merged commit 93d6325 into cbrnr:main Feb 13, 2025
16 checks passed
@cbrnr
Copy link
Owner

cbrnr commented Feb 13, 2025

Thanks @simon-p-2000!

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.

3 participants