This repository contains the PyTorch implementation of the GRU-D model described in the paper
Recurrent Neural Networks for Multivariate Time Series with Missing Values
by Che et al. The model is used to perform a mortality prediction task on the PhysioNet 2012 dataset.
- python 3.5
- torch 0.1.12
Please report any issues to me [email protected].
- Zhengping Che, Sanjay Purushotham, Kyunghyun Cho, David Sontag & Yan Liu. Recurrent Neural Networks for Multivariate Time Series with Missing Values . SREP (2018)
- PhysioNet 2012 dataset