pycaus is a python package for causal survival analysis and counterfactual classaification predection with PyTorch, built on the torchtuples package for training PyTorch models.
Note: This package is still in its early stages of development..
The package contains implementations of ..
To get started you first need to install PyTorch. You can then install pycox via pip:
pip install pycausThe package only works for python 3.6+.
[1] Ayoub Abraich, Agathe Guilloux and Blaise Hanczar. SurvCaus : Representation Balancing for Survival Causal Inference. Preprint Arxiv, 2022. [paper]