Diff of /README.md [000000] .. [a23a6e]

Switch to unified view

a b/README.md
1
# SALMON 🐟
2
SALMON: Survival Analysis Learning with Multi-Omics Neural Networks
3
4
### SALMON architecture with the implementation of Cox proportional hazards regression networks:
5
![SALMON Architecture](figures/Figure1.png)
6
7
### Publication:
8
Huang, Zhi, et al. "SALMON: Survival Analysis Learning with Multi-Omics Neural Networks on Breast Cancer." Frontiers in Genetics 10 (2019): 166. [[Link]](https://www.frontiersin.org/articles/10.3389/fgene.2019.00166/abstract)
9
10
11
## 1. Installation
12
* Download Anaconda from https://www.anaconda.com/download/#linux/ with python 3 version.
13
* Install Pytorch, tqdm, imblearn, and lifelines.
14
15
```bash
16
conda install pytorch torchvision -c pytorch
17
conda install -c conda-forge tqdm
18
conda install -c conda-forge imbalanced-learn
19
conda install -c conda-forge lifelines
20
```
21
22
### Alternative installation
23
To install the libraries with the specified version, `requirements.txt` can be used as a configuration file with the following command:
24
25
```bash
26
pip install -r requirements.txt
27
```
28
29
## 2. Run the main routine
30
31
32
```bash
33
cd experiments
34
python main.py
35
```