causal-tree-learn 1.0.1

Python implementation of causal trees with validation

CTL

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

Requirements

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

Installation

pip install causal_tree_learn

Demo Code

Two demo codes are available to run.

python binary_example.py

Runs the tree on a binary example (asthma.txt)

python trigger_example.py

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)