huangzhii / SALMON

SALMON: Survival Analysis Learning with Multi-Omics Neural Networks
https://huangzhii.github.io/SALMON/
67 stars 22 forks source link
co-expression deep-learning machine-learning neural-networks survival-analysis

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