Warning: This package is now deprecated.
This package is now deprecated and we do not plan to maintain it going forward. We suggest one of the following alternatives:
eventStudy: Perform an Event Study in R
Created by David Novgorodsky and Bradley Setzler at the University of Chicago
The methods are explained in our companion Practical Guide to Event Studies.
Disclaimer: By using this software, you accept the terms of the MIT license. This is a work in progress. It is updated frequently. Please let us know if you find any bugs or have questions that are not addressed in the documentation.
Overview
Purpose
eventStudy
is an R package for performing event studies. It has many capabilites:
- It maximizes sample size by using a "stacked" approach to match all possible control observations to any given treatment observation at any event time;
- It allows the estimated treatment effects to vary over time and by treatment cohort. It can also impose a common treatment effect across cohorts (extending results by Abraham & Sun, 2018);
- It has built-in tools to correct for anticipation (extending results by Fadlon & Nielsen, 2019) and deviations from parallel trends (either using covariate balancing in the spirit of Abadie, 2005, or modeling parametric deviations from parallel trends).
Installation
Preferred method: The latest version of the package can be installed with the command devtools::install_github("setzler/eventStudy/eventStudy")
.
Installation without internet access: Clone this repo to your personal machine, use the command line approach of R CMD BUILD eventStudy/
to build the file eventStudy_0.1.0.tar.gz, move this .tar.gz file to the location without internet access, then install on that location using the command install.package("eventStudy_0.1.0.tar.gz",repo=NULL)
within an R session. In case you cannot clone and build the repo, we provide a (possibly out of date) version of the .tar.gz file above.
ES
command
The The event study is performed by the ES
command. The examples below show how to use it.
ES
command:
Required Arguments of the These required arguments simply supply ES
your data and tell ES
the names of your variables:
long_data
: This is the data in long format (e.g., each row corresponds to a household in a given calendar year). It must be a data.table.outcomevar
: This is the outcome of interest. It must be acharacter
.unit_var
: This is the individual. It must be acharacter
.cal_time_var
: This is the calendar time variable (e.g., year). It must be acharacter
.onset_time_var
: This is the time at which the observation is treated. It must be acharacter
.cluster_vars
: These are the variables used to cluster standard errors (e.g., setting it to the same variable asunit_var
is a common choice). It must be acharacter vector
.
ES
command:
Optional Arguments of the These optional arguments allow you to customize ES
to deal with issues like anticipation and non-parallel trends:
omitted_event_time
: This lets you decide the pre-treatment event time to use as the reference year. It must be an integer that is less than or equal to-1
. The default is-2
.min_control_gap
andmax_control_gap
:min_control_gap
is the minimum number of time periods ahead that a cohort must receive treatment in order to be included in the control group (e.g., Fadlon & Nielsen, 2019, usemin_control_gap=5
). It must be an integer that is at least1
. The default is1
.max_control_gap
is similar but for the maximum number of time periods ahead (e.g., Fadlon & Nielsen, 2019, also usemax_control_gap=5
). It must be an integer that is at least as large asmin_control_gap
. The default isInf
.anticipation
: This drops observations from the control group this many time periods ahead of the control groups receiving the treatment. For example, consider a treatment group that receives treatment in 2001. Ifmin_control_gap=5
andanticipation=0
, then another cohort treated in 2006 can be used as a control group up to 2005, but ifanticipation=1
, then they can be used as a control group up to 2004.homogeneous_ATT
: This will only estimate the pooled (homogeneous) treatment effects if set toTRUE
, and skip estimating the cohort-specific effects, which may be slow for large data. Default isFALSE
.never_treat_action
: This determines how to handle never-treated groups. Those who do not receive treatment should be coded asonset_time_var=NA
. The default option isnever_treat_action="none"
, in which case, it is expected that all observations are treated, and an error is thrown if any observations haveonset_time_var=NA
. Other options are:never_treat_action="only"
, in which case only never-winners are used as control groups;never_treat_action="exclude"
, in which case never-winners are never used as controls; andnever_treat_action="keep"
, in which case both never-winners and not-yet-winners are used as controls.control_subset_var
andcontrol_subset_event_time
: These variables allow one to require that the control group satisfies a logical condition at a given event time.control_subset_var
must be the name of a logical variable in the data (all values of this variable areTRUE
orFALSE
) andcontrol_subset_event_time
is an integer which specifies the event time at which the control group must have a value ofTRUE
oncontrol_subset_var
. The default iscontrol_subset_var=NA
, which bypasses this option.fill_zeros
:fill_zeros
is a logical indicator (TRUE
orFALSE
) which results in the data being filled and the outcome set to zero for all observations on a unit that are missing within the time frame included in the data. The default isFALSE
.linearize_pretrends
:linearize_pretrends
is a logical indicator that, when set toTRUE
, results in linear pre-trends being fit during the pre-event years, and then residualized out of both the pre-event and post-event outcomes. The default isFALSE
.residualize_covariates
: Likelinearize_pretrends
, but it residualizes on provided covariates rather than time. Eitherdiscrete_covars
(discrete covariates) orcont_covars
(continuous covariates) or both must be specified ascharacter vectors
of variable names. Discrete covariates will be residualized using fixed effects, while continuous covariates will be residualized linearly. The default isFALSE
.
ES
Plotting the results from You can make your own plots with the results. As a convenience, we have prepared these functions to automatically make your plots:
ES_plot_levels
will plot the levels across event times for the treated and control cohorts.ES_plot_ATTs
will plot the treatment effects across event times. By default, it will plot heterogeneous effects for each cohort; sethomogeneous_ATT = TRUE
to only plot the pooled effect under the homogeneity assumption.
Examples
Set-up
devtools::install_github("setzler/eventStudy/eventStudy")
library(eventStudy)
library(data.table)
library(ggplot2)
Let's simulate some data using our built-in convenience function called ES_simulate_data
:
# simulate the data with 1000 individuals (use only the first element)
sim_data <- ES_simulate_data(units = 1000)[["observed"]]
# view the simulated data
sim_data[]
## individual year treatment_year outcome
## 1: 1 1999 2003 0.8141879
## 2: 1 2000 2003 0.8703408
## 3: 1 2001 2003 0.7651807
## 4: 1 2002 2003 0.9349418
## 5: 1 2003 2003 0.4439295
## ---
## 6996: 1000 2001 2003 1.0030966
## 6997: 1000 2002 2003 0.7616094
## 6998: 1000 2003 2003 0.8115637
## 6999: 1000 2004 2003 0.5377100
## 7000: 1000 2005 2003 0.4799415
In this data, the treatment is received in the year given by the treatment_year
variable. The other variables are individual
, year
, and outcome
. We wish to perform an event study to understand the effect of this treatment on this outcome.
To get a sense of what is in the simulated data, here is a plot of the mean outcomes by year at which treatment is received:
# calculate the means
cohort_means <- sim_data[,list(outcome = mean(outcome)), list(treatment_year,year)]
# plot the means
ggplot(data=cohort_means,aes(x=year,y=outcome,colour=factor(treatment_year))) + geom_line() +
labs(x = "Year", y = "Outcome", color = "Treatment") + theme_bw(base_size=16)
(Note: even though the data is drawn from the model with perfect control groups, the control groups do not look great visually!)
Example 1. Basic Estimation
Given this data, we perform the basic event study as follows:
# run the event study
results <- ES(long_data=sim_data, outcomevar="outcome",
unit_var="individual", cal_time_var="year",
onset_time_var="treatment_year", cluster_vars="individual")
## INFO [2019-02-11 14:27:24] Beginning ES.
## INFO [2019-02-11 14:27:24] Beginning data stacking.
## INFO [2019-02-11 14:27:43] Successfully produced a stacked dataset with 27,992 rows.
## INFO [2019-02-11 14:27:44] Estimated heterogeneous case with OLS.
## INFO [2019-02-11 14:27:44] Estimated homogeneous case with OLS.
## INFO [2019-02-11 14:27:44] ES is finished.
Example 2. Estimation using only control cohorts that receive treatment at certain times in the future
Let's only pair a control observation to a treatment observation if the control observation is treated at least 3 years later (min_control_gap = 3
) and at most 5 years later (max_control_gap = 5
), and the year prior to their treatment may have anticipation so we set anticipation=1
to exclude the year prior to treatment as control years:
# run the event study
results2 <- ES(long_data=sim_data, outcomevar="outcome",
unit_var="individual", cal_time_var="year",
onset_time_var="treatment_year", cluster_vars="individual",
min_control_gap = 3, max_control_gap = 5, anticipation = 1)
## INFO [2019-02-11 14:27:44] Beginning ES.
## INFO [2019-02-11 14:27:44] Beginning data stacking.
## INFO [2019-02-11 14:27:54] Successfully produced a stacked dataset with 11,834 rows.
## INFO [2019-02-11 14:27:55] Estimated heterogeneous case with OLS.
## INFO [2019-02-11 14:27:55] Estimated homogeneous case with OLS.
## INFO [2019-02-11 14:27:55] ES is finished.
Example 3. Estimation using only never-winners as the control group
Let's generate some never-winners by re-coding some of the onset times as missing:
# coding some observations as never-winners
sim_data2 <- copy(sim_data)
unit_var = "individual"
onset_time_var = "treatment_year"
sim_data2[, recode := runif(1, min = 0, max = 1), list(get(unit_var))]
sim_data2[recode < 0.10, (onset_time_var) := NA]
# run the event study
results3 <- ES(long_data=sim_data2, outcomevar="outcome",
unit_var="individual", cal_time_var="year",
onset_time_var="treatment_year", cluster_vars="individual",
never_treat_action = "only")
## INFO [2019-02-11 14:27:55] Beginning ES.
## INFO [2019-02-11 14:27:55] Beginning data stacking.
## INFO [2019-02-11 14:28:15] Successfully produced a stacked dataset with 17,148 rows.
## INFO [2019-02-11 14:28:15] Estimated heterogeneous case with OLS.
## INFO [2019-02-11 14:28:16] Estimated homogeneous case with OLS.
## INFO [2019-02-11 14:28:16] ES is finished.
Example 4. Estimation using only control cohorts that satisfy a condition at an event time
Let's randomly assign some control groups to be valid in some years:
# adding the valid_control variable
sim_data[, valid_control := runif(nrow(sim_data)) < .5]
Now, with the control_subset_var
and control_subset_event_time
arguments, we can only use the control groups that satisfy valid_control == TRUE
in the year before the treatment group receives treatment:
# run the event study
results4 <- ES(long_data=sim_data, outcomevar="outcome",
unit_var="individual", cal_time_var="year",
onset_time_var="treatment_year", cluster_vars="individual",
control_subset_var="valid_control", control_subset_event_time=-1)
## INFO [2019-02-11 14:28:16] Beginning ES.
## INFO [2019-02-11 14:28:16] Beginning data stacking.
## INFO [2019-02-11 14:28:36] Successfully produced a stacked dataset with 19,478 rows.
## INFO [2019-02-11 14:28:36] Estimated heterogeneous case with OLS.
## INFO [2019-02-11 14:28:37] Estimated homogeneous case with OLS.
## INFO [2019-02-11 14:28:37] ES is finished.
Example 5. Removing linear deviations from parallel trends
Suppose we think that pre-trends differ from parallel in a linear way. Using linearize_pretrends = TRUE
will estimate these linear trends on pre-event years and extrapolate the trends to post-event years:
# run the event study
results5 <- ES(long_data=sim_data, outcomevar="outcome",
unit_var="individual", cal_time_var="year",
onset_time_var="treatment_year", cluster_vars="individual",
linearize_pretrends = TRUE)
## INFO [2019-02-11 14:28:37] Beginning ES.
## INFO [2019-02-11 14:28:37] Linearizing pre-trends.
## INFO [2019-02-11 14:28:37] Beginning data stacking.
## INFO [2019-02-11 14:28:57] Successfully produced a stacked dataset with 27,992 rows.
## INFO [2019-02-11 14:28:57] Estimated heterogeneous case with OLS.
## INFO [2019-02-11 14:28:58] Estimated homogeneous case with OLS.
## INFO [2019-02-11 14:28:58] ES is finished.
Example 6. Residualzing on covariates where some are discrete and some are continuous
Let's add some discrete and continuous covariates to the data:
# adding some covariates
sim_data[, cont_covar1 := runif(nrow(sim_data))]
sim_data[, cont_covar2 := runif(nrow(sim_data))]
sim_data[, discrete_covar1 := as.integer(runif(nrow(sim_data)) < .5)] # 2 levels
sim_data[, discrete_covar2 := as.integer(runif(nrow(sim_data)) < .5) + as.integer(runif(nrow(sim_data)) < .5)] # 3 levels
By default, each level of a discrete covariate is given its own fixed effect (except an omitted level), while continuous covariates are assumed to enter linearly. Be careful not to include too many levels in the discrete covariates since R can become very slow when estimating a large number of fixed effects. The coefficients are estimated on pre-treatment years and then extrapolated to the post-treatment years, allowing for the covariates to change in the meantime.
The argument residualize_covariates = TRUE
will tell ES
to residualize on covariates. The discrete covariates are passed to ES
with the argument discrete_covars
, while the continuous covariates are passed with the argument cont_covars
:
# run the event study
results6 <- ES(long_data=sim_data, outcomevar="outcome",
unit_var="individual", cal_time_var="year",
onset_time_var="treatment_year", cluster_vars="individual",
residualize_covariates = TRUE,
discrete_covars = c('discrete_covar1','discrete_covar2'),
cont_covars = c('cont_covar1','cont_covar2'))
## INFO [2019-02-11 14:28:58] Beginning ES.
## INFO [2019-02-11 14:28:58] Residualizing on covariates.
## INFO [2019-02-11 14:28:58] Beginning data stacking.
## INFO [2019-02-11 14:29:18] Successfully produced a stacked dataset with 27,992 rows.
## INFO [2019-02-11 14:29:19] Estimated heterogeneous case with OLS.
## INFO [2019-02-11 14:29:19] Estimated homogeneous case with OLS.
## INFO [2019-02-11 14:29:19] ES is finished.
Example 7. Filling in missing rows with zeros.
Let's randomly drop some of the rows from the data so that it is no longer balanced:
sim_data <- sim_data[runif(nrow(sim_data)) < .8]
Suppose that missing rows have a true value of zero for the outcome. We can add the missing rows and fill in zeros for the outcome using the fill_zeros = TRUE
argument:
# run the event study
results7 <- ES(long_data=sim_data, outcomevar="outcome",
unit_var="individual", cal_time_var="year",
onset_time_var="treatment_year", cluster_vars="individual",
fill_zeros = TRUE)
## INFO [2019-02-11 14:29:19] Beginning ES.
## INFO [2019-02-11 14:29:19] Filling in zeros.
## INFO [2019-02-11 14:29:19] Beginning data stacking.
## INFO [2019-02-11 14:29:39] Successfully produced a stacked dataset with 27,992 rows.
## INFO [2019-02-11 14:29:40] Estimated heterogeneous case with OLS.
## INFO [2019-02-11 14:29:40] Estimated homogeneous case with OLS.
## INFO [2019-02-11 14:29:40] ES is finished.
Note: This cannot be combined with residualizing on covariates, since ES would need to know how to fill in the covariates during the years that observations are missing.
ES
Bonus Example. Plotting the results from Now, we plot the results. First, we plot the treatment and control means. ES
has constructed the appropriate control group for each treatment group. We can see that it looks much cleaner than in the raw data:
ES_plot_levels(results, lower_event = -3, upper_event = 5) + ylab("Mean of the Outcome")
Note that, becuase omitted_event_time=-2
, the treatment and control group means must both be zero at event time -2, for each treatment-control pair.
Next, we plot the treatment effects, comparing the cohort-specific effects to the pooled effect that comes from imposing homogeneity across cohorts:
ES_plot_ATTs(results, lower_event = -3, upper_event = 5) + ylab("ATT Estimate (95% CI)")
Finally, if we only care about the estimates under the homogeneity assumption, we can show these as follows:
ES_plot_ATTs(results, lower_event = -3, upper_event = 5, homogeneous_ATT = TRUE) +
ylab("ATT Estimate (95% CI)")