TorchSurv: A Lightweight Package for Deep Survival Analysis

arXiv:2404.10761v1 Announce Type: new
Abstract: TorchSurv is a Python package that serves as a companion tool to perform deep survival modeling within the PyTorch environment. Unlike existing libraries that impose specific parametric forms, TorchSurv enables the use of custom PyTorch-based deep survival mod- els. With its lightweight design, minimal input requirements, full PyTorch backend, and freedom from restrictive survival model parameterizations, TorchSurv facilitates efficient deep survival model implementation and is particularly beneficial for high-dimensional and complex input data scenarios

abstract analysis arxiv companion cs.lg design environment forms libraries modeling package parametric python pytorch requirements survival tool type