Implementation of DeepSurv using Keras
Project description
DeepSurvK
Implementation of DeepSurv using Keras
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!
:bookmark_tabs: Documentation
You can find the complete package's documentation here.
: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
: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:
- Katzman, Jared L., et al. "DeepSurv: personalized treatment recommender system using a Cox proportional hazards deep neural network." BMC medical research methodology 18.1 (2018): 24. [BibTeX]
- Arturo Moncada-Torres. DeepSurvK. Accessed on [MONTH, 20XX].
: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
Built Distribution
Hashes for deepsurvk-0.1.0-py2.py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 0d7c6582fa047b099d52c7af63ec8f2c3c7c806e1ebbebc9cee97383c7b3ad8e |
|
MD5 | 4aa15f488842d4839be57266203280f6 |
|
BLAKE2b-256 | 98b24d1858096775a5ebd95a007cc419b43e61b13121f176d7a8a3196bf6187c |