Skip to main content

Implementation of DeepSurv using Keras

Project description

Implementation of DeepSurv using Keras

PyPI version Build Status Documentation PyUp

MotivationFeaturesDocumentationLicenseReferencesCredits


: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.

Files for deepsurvk, version 0.2.2
Filename, size File type Python version Upload date Hashes
Filename, size deepsurvk-0.2.2-py2.py3-none-any.whl (22.4 kB) File type Wheel Python version py2.py3 Upload date Hashes View
Filename, size deepsurvk-0.2.2.tar.gz (115.1 kB) File type Source Python version None Upload date Hashes View

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring DigiCert DigiCert EV certificate Facebook / Instagram Facebook / Instagram PSF Sponsor Fastly Fastly CDN Google Google Object Storage and Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Salesforce Salesforce PSF Sponsor Sentry Sentry Error logging StatusPage StatusPage Status page