CausalForge
CausalForge is a Python package that provides a suite of modeling & causal inference methods using machine learning algorithms based on Elevence Health recent research. It provides convenient APIs that allow to estimate Propensity Score, Average Treatment Effect (ATE), Conditional Average Treatment Effect (CATE) or Individual Treatment Effect (ITE) from experimental or observational data. Methods have been redesigned for production. Check out the documentation.
Installing Python Package
We recommend to create a proper enviroment with tensorflow and pytorch installed. For example, for a local Mac enviroment without GPUs:
conda env create -f env_mac.yml
conda activate causalforge
You can install it after cloning this repository, i.e.
git clone https://github.com/anthem-ai/causalforge
cd causalforge
[sudo] pip install -e . [--trusted-host pypi.org --trusted-host files.pythonhosted.org]
or directly from the repository (development), i.e.
pip install --upgrade git+https://github.com/anthem-ai/causalforge [--trusted-host pypi.org --trusted-host files.pythonhosted.org]
or directly from PyPI, i.e.
pip install causalforge
After installing you can import classes and methods, e.g.
import causalforge
causalforge.__version__
Testing
cd tests
pytest --disable-warnings
Citation
@article{tesei2023learning,
title={Learning end-to-end patient representations through self-supervised covariate balancing for causal treatment effect estimation},
author={Tesei, Gino and Giampanis, Stefanos and Shi, Jingpu and Norgeot, Beau},
journal={Journal of Biomedical Informatics},
volume={140},
pages={104339},
year={2023},
publisher={Elsevier}
}
1 maintainer
Release files for causalforge 0.1.0
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| causalforge-0.1.0.tar.gz | 26.3 kB | Details |
Release files / causalforge-0.1.0.tar.gz
| Download URL | causalforge-0.1.0.tar.gz |
|---|---|
| Size | 26.3 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
146d5ac0498a9892c44750d8a0f14cf86e960b402e22e1f80349e1ee42ceebdc
|
|
BLAKE2b-256 checksum How to use checksums |
6937aff0102eb36b0ff7294063d4a4cdadb061eb87e188d02ee8a382132c0e25
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/4.0.2 CPython/3.9.7
|
