Add causal trees interpretation example (#536)

alexander-pv · web-flow · commit d7473d3b58b9 · 2022-08-22T12:19:51.000-07:00
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
