Python implementation of causal trees with validation
Christopher Tran, Elena Zheleva, "Learning Triggers for Heterogeneous Treatment Effects", AAAI 2019.
Our method is based on and adapted from: https://github.com/susanathey/causalTree
- Python 3
- graphviz (if you want to plot the tree)
pip install causal_tree_learn
or clone the repository
python setup.py build_ext --inplace
Two demo codes are available to run.
Runs the tree on a binary example (asthma.txt)
Runs a tree on a trigger problem where the treatment is continuous (note for now the example is made up and treatment does not affect outcome, this is only to show example code)
Release history Release notifications | RSS feed
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Hashes for causal_tree_learn-2.42-cp310-cp310-macosx_12_0_arm64.whl
Hashes for causal_tree_learn-2.42-cp37-cp37m-macosx_10_9_x86_64.whl