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
through pip
pip install causal_tree_learn
or clone the repository
python setup.py build_ext --inplace
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)
Release files for causal-tree-learn 2.43
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| causal-tree-learn-2.43.tar.gz | 121.6 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| causal_tree_learn-2.43-cp310-cp310-macosx_12_0_arm64.whl | CPython 3.10 | CPython 3.10 | macOS 12.0+ ARM64 | Details |
Total release size: 338.5 kB
Release files / causal-tree-learn-2.43.tar.gz
| Download URL | causal-tree-learn-2.43.tar.gz |
|---|---|
| Size | 121.6 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
0807f314ae8645f60725f5ecae7aee0ede8b365b7f628fce9a32e9c0b7553c96
|
|
BLAKE2b-256 checksum How to use checksums |
c8fd491e078adb7ce1b32c25340296c9cea51c24aad1443dc6ed0036bf3112ee
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/4.0.1 CPython/3.10.6
|
Release files / causal_tree_learn-2.43-cp310-cp310-macosx_12_0_arm64.whl
| Download URL | causal_tree_learn-2.43-cp310-cp310-macosx_12_0_arm64.whl |
|---|---|
| Size | 216.9 kB |
| Tags | CPython 3.10 macOS 12.0+ ARM64 |
|
SHA-256 checksum How to use checksums |
f089b99701b50d6dc059d0ae44e7180a18d93a9bed281794f935b5fdf196c95c
|
|
BLAKE2b-256 checksum How to use checksums |
276df13117f3b40c95b8cdf350a268bfece7e59ef52b338f108630b0f689bad3
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/4.0.1 CPython/3.10.6
|