Skip to main content

Implementation of DeepSurv using Keras

Project description

Implementation of DeepSurv using Keras

PyPI version Build Status Documentation PyUp


:pray: Motivation

DeepSurv is a Cox Proportional Hazards deep neural network used for modeling interactions between a patient's covariates and treatment effectiveness. It was originally proposed by Katzman et. al (2018) and implemented in Theano (using Lasagne).

Unfortunately, Theano is no longer supported. There have been some attempts in recreating DeepSurv in other DL platforms, such as czifan's DeepSurv.pytorch. However, given its popularity and ease of use, I think TensorFlow 2's Keras is a great option for this task.

mexchy1000 created DeepSurv_Keras. However, it is a very raw prototype: it is not properly documented nor validated. Moreover, it is not being actively supported anymore. Therefore, I used it as a rough starting point for the development of DeepSurvK.

This is my first Python package. I am sure there are many places where it could be improved. Feedback is always welcome!

:tada: Features

  • Implemented using Keras (using TensorFlow 2)
  • Includes the original datasets together with a proper description of the variables
  • Designed with data as pandas DataFrames in mind
  • Visualization tools for the most common plots for fast and easy exploration and prototyping
  • Treatment recommender
  • (Basic) hyperparameter optimization using grid and randomized search

:bookmark_tabs: Documentation

You can find the complete package's documentation here. Unfortunately, I haven't had as much time as I would like to work on it. Alternatively, I strongly recommend you take look at the example notebooks.

:page_with_curl: License

This package uses the MIT license

:black_nib: References

If you are using DeepSurvK, please cite the original DeepSurv paper, as well as the current repository as follows:

:label: Credits

This package was developed in Spyder (a fantastic open-source Python IDE) using Cookiecutter and the arturomoncadatorres/cookiecutter-pypackage project template.

Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

deepsurvk-0.2.2.tar.gz (115.1 kB view hashes)

Uploaded Source

Built Distribution

deepsurvk-0.2.2-py2.py3-none-any.whl (22.4 kB view hashes)

Uploaded Python 2 Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page