causal_tree_test.rmd

---
title: "causal_trees_demo"
output: html_document
date: '2022-11-09'
---

```{r setup, include=FALSE}
knitr::opts_chunk$set(echo = TRUE)
```


```{r}

library(causalTree)
library(lmtest)
library(sandwich)
library(ggplot2)

marketing <- read.csv("causal_trees_data.csv")

tree <- causalTree(totalPurchases ~ SMSRecipient + emailRecipient + previousCustomer + juneVisit + as.factor(store),
                   data = marketing, 
                   treatment = marketing$treatment, 
                   split.Rule = "CT", 
                   cv.option = "CT", 
                   split.Honest = T, 
                   cv.Honest = T, 
                   split.Bucket = F, 
                   minsize = 5)
                  
opcp <- tree$cptable[,1][which.min(tree$cptable[,4])]

opfit <- prune(tree, opcp)

rpart.plot(opfit)

```


```{r}
tree2 <- causalTree(totalPurchases ~ SMSRecipient + emailRecipient + previousCustomer + juneVisit + as.factor(store), 
                    data = marketing, 
                    treatment = marketing$treatment, 
                    split.Rule = "CT", 
                    cv.option = "CT", 
                    split.Honest = F, 
                    cv.Honest = F, 
                    split.Bucket = F, 
                    minsize = 5)

opcp <- tree2$cptable[,1][which.min(tree2$cptable[,4])]
opfit <- prune(tree2, opcp)
rpart.plot(opfit)

```

causal_trees_data.csv