Skip to main content

Python implementation of causal trees with validation

Project description


Christopher Tran, Elena Zheleva, "Learning Triggers for Heterogeneous Treatment Effects", AAAI 2019.

Our method is based on and adapted from:


  • Python 3
  • sklearn
  • scipy
  • graphviz (if you want to plot the tree)


through pip

pip install causal_tree_learn

or clone the repository

python build_ext --inplace

Demo Code

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)

Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Files for causal-tree-learn, version 2.41
Filename, size File type Python version Upload date Hashes
Filename, size causal_tree_learn-2.41-cp37-cp37m-macosx_10_9_x86_64.whl (231.5 kB) File type Wheel Python version cp37 Upload date Hashes View
Filename, size causal_tree_learn-2.41.tar.gz (122.9 kB) File type Source Python version None Upload date Hashes View

Supported by

Pingdom Pingdom Monitoring Google Google Object Storage and Download Analytics Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN DigiCert DigiCert EV certificate StatusPage StatusPage Status page