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

new feature: torch dataset #71

Merged
merged 20 commits into from
Dec 14, 2024
Merged

new feature: torch dataset #71

merged 20 commits into from
Dec 14, 2024

Conversation

eroell
Copy link
Contributor

@eroell eroell commented Dec 10, 2024

Introduce the binned time-series of ehrdata.r as a pytorch dataset, streamed directly from the RDBMS without needing to load this potentially large datastructure into memory at any point

Tensor Dataset

  • New class ehrdata.tl.EHRDataset, subclassing pytorch dataset
  • Notebook for how to use EHRDataset
  • Basic tests

Misc

  • Construct .t in setup_variables and setup_interval_variables directly from RDBMS
  • Construct .var in setup_variables and setup_interval_variables directly from RDBMS
  • Unit checks in in setup_variables directly from RDBMS
  • Add tests for unit checks

Copy link

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

@eroell eroell marked this pull request as ready for review December 14, 2024 21:18
@eroell eroell merged commit 80acdfd into main Dec 14, 2024
6 checks passed
@eroell eroell deleted the feature/omop-torch-dataset branch December 14, 2024 21:19
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