CTL

Please find the new repository on our lab github where all new updates will be: https://github.com/edgeslab/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)

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)