shihux/sa_transformer

Code for "Transformer-Based Deep Survival Analysis"

Transformer-Based Deep Survival Analysis

The PyTorch implementation of the paper: Transformer-Based Deep Survival Analysis.

Requirements

This code is tested on:

• Python 3.6
• PyTorch 1.1.0
• Nvidia TITAN Xp GPU