plot.emfx

[grantmcdermott]

Closes #53

Examples borrowed from the updated vignette.

library(etwfe)

data("mpdta", package = "did")

mod_es2 = etwfe(
  lemp ~ lpop, tvar = year, gvar = first.treat, data = mpdta, vcov = ~countyreal,
  cgroup = "never" # <= use never treated as control group
) |>
  emfx(type = "event")

plot(mod_es2)

# args passed to tinyplot
plot(
  mod_es2,
  type = "ribbon",
  col  = "darkcyan",
  xlab = "Years post treatment",
  main = "Minimum wage effect on (log) teen employment",
  sub  = "Note: Using never-treated as control group"
)

#
# Fancier heterogeneous treatment effects example 
#
gls_fips = c("IL" = 17, "IN" = 18, "MI" = 26, "MN" = 27,
             "NY" = 36, "OH" = 39, "PA" = 42, "WI" = 55)
mpdta$gls = substr(mpdta$countyreal, 1, 2) %in% gls_fips
etwfe(
  lemp ~ lpop, tvar = year, gvar = first.treat, data = mpdta,
  cgroup = "never",
  vcov = ~countyreal,
  xvar = gls           ## <= het. TEs by gls
) |>
  emfx("event") |>
  plot()

^{Created on 2024-12-16 with reprex v2.1.1}