/pytorch-retain

pytorch-retain

Primary LanguagePythonGNU General Public License v3.0GPL-3.0

pytorch-retain

This is PyTorch re-implementation of the following paper:

Choi, Edward, Mohammad Taha Bahadori, Jimeng Sun, Joshua Kulas, Andy Schuetz, and Walter Stewart. 2016. “RETAIN: An Interpretable Predictive Model for Healthcare Using Reverse Time Attention Mechanism.” In Advances in Neural Information Processing Systems 29, edited by D. D. Lee, M. Sugiyama, U. V. Luxburg, I. Guyon, and R. Garnett, 3504–12. Curran Associates, Inc.