41eed00 2020-08-26 Update README.md
198bb23 2020-04-10 Merge pull request #39 from jmigual/master
888f95c 2020-04-10 Merge pull request #62 from piyush-kgp/master
27af52f 2020-04-10 import sys
f4f3186 2020-04-10 import sys
d2bab01 2020-04-10 python3 dictionary doesnt have haskey attribute
27883dc 2018-09-22 Merge pull request #46 from leihuang/patch-1
f62ccee 2018-09-21 Update __init__.py
ef32844 2018-09-21 Update __init__.py
a924363 2018-09-21 Update deep_surv.py
d13b838 2018-03-12 Fix typo in example code
3c20b35 2018-01-21 Merge revisions

# DeepSurv Experiments

Experiments are run using Docker containers built off of the [floydhub](https://github.com/floydhub/dl-docker) deep learning Docker images. 

### Requirements

Download and install [Docker](https://www.docker.com/community-edition#/download). If you plan to use FloydHub's GPU tag, install [nvidia-docker](https://github.com/NVIDIA/nvidia-docker).
	
### Experiments

To run one of the experiments from the paper, use the following command from this directory:

	 EXPERIMENT=${EXPERIMENT_ID} docker-compose up --build

The following experiments are provided:

| Experiment    | `${EXPERIMENT_ID} `|
| ------------- |:-------------:|
| Simulated Linear Data | `linear` |
| Simulated Nonlinear (Gaussian) Data | `gaussian` |
| Worchester Heart Attack Study (WHAS) | `whas` |
| Study to Understand Prognoses Preferences Outcomes and Risks of Treatment (SUPPORT) | `support` |
| Molecular Taxonomy of Breast Cancer International Consortium (METABRIC) | `metabric` |
| Simulated Treatment Data | `treatment` |
| Rotterdam & German Breast Cancer Study Group (GBSG) | `gbsg` |

The hyper-parameters for each experiment are in this repo's directory: `DeepSurv/experiments/deepsurv/models/`

For more details on each experiment, reference [DeepSurv: Personalized Treatment Recommender System Using A Cox Proportional Hazards Deep Neural Network](http://arxiv.org/abs/1606.00931).

#### Using GPU

If you have nvidia-docker installed, you can run DeepSurv experiments using your GPU. To do so change the tag in the first line of your experiment's docker file. 

For example, to run the simulated linear data experiment with the GPU change the first line the file `./deepsurv/Docker.linear` to:

	FROM floydhub/dl-docker:gpu	

experiments/deepsurv:
Dockerfile.gaussian
Dockerfile.gbsg
Dockerfile.linear
Dockerfile.metabric
Dockerfile.support
Dockerfile.treatment
Dockerfile.whas
models

experiments/deepsurv/models:
gaussian_model_0.json
gaussian_model_selu_revision.0.json
gaussian_model_selu_revision.1.json
gbsg_model_selu_revision.0.json
gbsg_selu_adam_model_0.json
linear_model_selu_0.json
linear_model_selu_revision.0.json
linear_model_selu_revision.json
metabric_IHC4_clinical_adam_0.json
metabric_model_selu_revision.0.json
support_model_selu_revision.0.json
support_selu_model_adam_0.json
treatment_adam_selu_model_0.json
treatment_model_selu_revision.0.json
whas_model_selu_0.json
whas_model_selu_revision.0.json

experiments/scripts:
cph_run.py
deepsurv_run.py
rsf_run.py
