大数据安全>> CTL>> 返回
项目作者： chris-tran-16

项目描述 ：
A causal tree method for heterogeneous treatment effects in Python
高级语言： Python
项目地址: git://github.com/chris-tran-16/CTL.git
创建时间： 2019-04-16T17:57:29Z
项目社区：https://github.com/chris-tran-16/CTL
开源协议：
下载


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.

  1. python binary_example.py

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

  1. 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)