VectorInstitute / odyssey

A toolkit for developing foundation models using Electronic Health Record (EHR) data.

https://vectorinstitute.github.io/EHRMamba

Apache License 2.0

26 stars 10 forks source link

Add new forecasting features and support for MEDS training #83

Closed Adibvafa closed 2 months ago

Adibvafa commented 2 months ago

PR Type

Feature

Short Description

This PR will enhance the forecasting functionalities of Odyssey and enable training with MEDS data format.

Adibvafa commented 2 months ago

@amrit110 PR is ready for review. The merge conflict for odyssey/models/ehr_mamba/mamba_utils.py is because the pre-commit still modfies the docstring of mamba related classes (they should end with Returns: not Returns: ----), could you fix this?