GRU-D

Description

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.

Requirements

• python 3.5
• torch 0.1.12

Issues

Please report any issues to me juiting.hsu@nyu.edu.

Reference

• 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