Introduction
A pytorch implementation of RETAIN, which is an interpretable predictive model for healthcare applications.
Please refer to the origin work for information on dataset and data preprocessing
Relevant Publications
RETAIN implements an algorithm introduced in the following paper:
RETAIN: An Interpretable Predictive Model for Healthcare using Reverse Time Attention Mechanism
Edward Choi, Mohammad Taha Bahadori, Joshua A. Kulas, Andy Schuetz, Walter F. Stewart, Jimeng Sun,
NIPS 2016, pp.3504-3512