Skip to content

Add GRU-D model for mortality predition.#986

Open
tracy030115 wants to merge 1 commit intosunlabuiuc:masterfrom
tracy030115:GRUD-model
Open

Add GRU-D model for mortality predition.#986
tracy030115 wants to merge 1 commit intosunlabuiuc:masterfrom
tracy030115:GRUD-model

Conversation

@tracy030115
Copy link
Copy Markdown

Contributor: Tracy Zhang (tracyz3@illinois.edu)
Contribution Type: Model (Reproducibility and ablation)
Paper: https://arxiv.org/abs/1908.00690
Description: Implemented GRU-D (Gated Recurrent Unit with Decay), a RNN designed for irregularly sampled clinical time series with missing values. GRU-D extends the standard GRU with two decay mechanisms: input decay and hidden state decay. The model compared Raw, CUI, and Clinical feature representations along with hyperparameter tuning and ablation on the two decay mechanisms.
Files to Review:

  • pyhealth/models/grud.py
  • pyhealth/models/__init__.py
  • tests/core/test_grud.py
  • docs/api/models/pyhealth.models.GRUD.rst
  • docs/api/models.rst
  • examples/mortality_prediction/mimic3_mortality_grud.ipynb

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