compbiomed-unito / survhive

Convenient, opinionated wrapper around some (deep) survival models
MIT License
2 stars 0 forks source link
machine-learning survival-analysis

SurvHive

SurvHive is a Python package that provides an interface to various survival-analysis models, from Cox to deep-learning-based ones, making it easier to access and compare different methods.

It is designed so that all method adapters are compliant scikit-learn estimators, allowing interoperability with scikit-learn facilities and it offers a range of features for model selection, parameter tuning, and performance evaluation.

Models provided:

From scikit-survival:

From Pycox:

From Auton Survival:

From LassoNet:

From SurvTRACE:

Metrics

SurvHive provides multiple metrics for models evaluation, such as Harrell C-index, Antolini score, Brier score, and AUROC, and allows for the creation of user-defined ones.

Installation

See here for instruction.

Getting Started

See our guide.

API Documentation

See doc/ folder

Cite

Submitted for publication

License

This code is MIT-licensed (see included LICENSE file). The wrapped codes are licensed according to their own terms (mostly MIT).