Tensor-based Temporal Multi-Task Survival Analysis

• This repository is an implementation of the tensor-based temporal multi-task survival analysis model proposed in our work published on TKDE'20: Tensor-based Temporal Multi-Task Survival Analysis.

Citation

Ping Wang, Tian Shi, and Chandan K. Reddy. Tensor-based Temporal Multi-Task Survival Analysis. IEEE Transactions on Knowledge and Data Engineering (TKDE), 2020.

@ARTICLE{wang2020tensor,  
author={Wang, Ping and Shi, Tian and Reddy, Chandan K.},  
journal={IEEE Transactions on Knowledge and Data Engineering},  
title={Tensor-based Temporal Multi-Task Survival Analysis},   
year={2020}}

Dataset

• Medical Information Mart for Intensive Care III (MIMIC III)
• Employee attrition dataset from IBM Watson Analytics

Usuage

• python main.py