SALMON 🐟

SALMON: Survival Analysis Learning with Multi-Omics Neural Networks

SALMON architecture with the implementation of Cox proportional hazards regression networks:

SALMON Architecture

Publication:

Huang, Zhi, et al. β€œSALMON: Survival Analysis Learning with Multi-Omics Neural Networks on Breast Cancer.” Frontiers in Genetics 10 (2019): 166. [Link]

1. Installation

conda install pytorch torchvision -c pytorch
conda install -c conda-forge tqdm
conda install -c conda-forge imbalanced-learn
conda install -c conda-forge lifelines

2. Run the main routine

cd experiments
python main.py