\begin{knitrout}
\definecolor{shadecolor}{rgb}{0.969, 0.969, 0.969}\color{fgcolor}\begin{kframe}
\begin{alltt}
\hlcom{#Computing the ATE}
\hlstd{Xm}\hlkwb{<-}\hlkwd{cbind}\hlstd{(}\hlnum{1}\hlstd{,X1,X2,X1}\hlopt{*}\hlstd{X2)}
\hlstd{ps.fit.drgee}\hlkwb{<-}\hlnum{1}\hlopt{/}\hlstd{(}\hlnum{1}\hlopt{+}\hlkwd{exp}\hlstd{(}\hlopt{-}\hlstd{Xm} \hlopt{%*%} \hlstd{alpha.hat))}
\hlstd{mu0.drgee}\hlkwb{<-}\hlkwd{mean}\hlstd{(beta.hat} \hlopt{+} \hlstd{(}\hlnum{0}\hlopt{-}\hlstd{ps.fit.drgee)}\hlopt{*}\hlstd{Xm}\hlopt{%*%}\hlstd{psi.hat}\hlopt{+}\hlstd{ps.fit.drgee}\hlopt{*}\hlstd{Xm}\hlopt{%*%}\hlstd{psi.prime.hat)}
\hlstd{mu1.drgee}\hlkwb{<-}\hlkwd{mean}\hlstd{(beta.hat} \hlopt{+} \hlstd{(}\hlnum{1}\hlopt{-}\hlstd{ps.fit.drgee)}\hlopt{*}\hlstd{Xm}\hlopt{%*%}\hlstd{psi.hat}\hlopt{+}\hlstd{ps.fit.drgee}\hlopt{*}\hlstd{Xm}\hlopt{%*%}\hlstd{psi.prime.hat)}
\hlkwd{c}\hlstd{(mu0.drgee,mu1.drgee,mu1.drgee}\hlopt{-}\hlstd{mu0.drgee)}
\end{alltt}
\begin{verbatim}
+ [1]  4.446842  2.537881 -1.908961
\end{verbatim}
\end{kframe}
\end{knitrout}