This repository is a Pytorch implementation of the paper : Dynamic Deep Hit. Original implementation in Tensorflow can be found at chl8856/Dynamic-DeepHit.
The code is based on the Deep Survival Machines repository and use a similar module structure.
All the code related to Dynamic Deep Hit is loacted in ddh. Only API should be used for application.
An example is available in the examples folder.
- Full documentation
- Further testing to ensure the model is reproducing the original paper's performance