From 079dee71e3b20084047b9b022f64ed6b430d3df8 Mon Sep 17 00:00:00 2001 From: Jared Katzman Date: Tue, 8 Aug 2017 22:46:45 -0400 Subject: [PATCH] Update README.md --- README.md | 11 +++++++++-- requirements.txt | 6 ++++++ 2 files changed, 15 insertions(+), 2 deletions(-) create mode 100644 requirements.txt diff --git a/README.md b/README.md index dd49d62..dfb4759 100644 --- a/README.md +++ b/README.md @@ -6,13 +6,13 @@ DeepSurv has an advantage over traditional Cox regression because it does not re DeepSurv can be used in numerous survival analysis applications. One medical application is provided: recommend_treatment, which provides treatment recommendations for a set of patient observations. -For more details, see full paper [Deep Survival: A Deep Cox Proportional Hazards Network](http://arxiv.org/abs/1606.00931). +For more details, see full paper [DeepSurv: Personalized Treatment Recommender System Using A Cox Proportional Hazards Deep Neural Network](http://arxiv.org/abs/1606.00931). ## Installation: ### Dependencies -Theano, Lasagne, lifelines, matplotlib (for visualization) and all of their respective dependencies. +Theano, Lasagne (bleeding edge version), lifelines, matplotlib (for visualization) and all of their respective dependencies. ### Installing @@ -62,3 +62,10 @@ You can then evaluate its success on testing data: If you have matplotlib installed, you can visualize the training and validation curves after training the network: deepsurv.plot_log(log) + +## Running Experiments + +Experiments are run using Docker containers built off of the [floydhub](https://github.com/floydhub/dl-docker) deep learning Docker images. DeepSurv can be run on either the CPU or the GPU with nvidia-docker. + +To run an experiment, define the experiment name as an environmental variable `EXPRIMENT`and run the docker-compose file. + \ No newline at end of file diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..d168d7b --- /dev/null +++ b/requirements.txt @@ -0,0 +1,6 @@ +h5py==2.7.0 +lifelines==0.9.4 +logger==1.4 +Optunity==1.1.1 +tensorboard-logger==0.0.3 +lasagne==0.2.dev1 \ No newline at end of file