Add causal trees interpretation example (#536)

uber · Aug 22, 2022 · d7473d3 · d7473d3
1 parent c82d636
commit d7473d3
Show file tree

Hide file tree

Showing 3 changed files with 677 additions and 1 deletion.
diff --git a/Makefile b/Makefile
@@ -1,7 +1,11 @@
 .PHONY: build_ext
-build_ext: clean
+build_ext: install_req clean
 	python setup.py build_ext --force --inplace
 
+.PHONY: install_req
+install_req:
+	pip install -r requirements.txt
+
 .PHONY: build
 build: build_ext
 	python setup.py bdist_wheel

diff --git a/causalml/inference/tree/causal/builder.pyx b/causalml/inference/tree/causal/builder.pyx
@@ -1,3 +1,9 @@
+# cython: cdivision=True
+# cython: boundscheck=False
+# cython: wraparound=False
+# cython: language_level=3
+# cython: linetrace=True
+
 from sklearn.tree._tree cimport Tree, TreeBuilder
 from sklearn.tree._tree cimport Splitter, SplitRecord
 from sklearn.tree._utils cimport StackRecord, Stack

diff --git a/examples/causal_trees_interpretation.ipynb b/examples/causal_trees_interpretation.ipynb