ipt: a Python 2.7 package for causal inference by inverse probability tilting ----------------------------------------------------------------------------- by Bryan S. Graham, UC - Berkeley, e-mail: bgraham@econ.berkeley.edu This package includes a Python 2.7 implementation of the Average Treatment Effect of the Treated (ATT) estimator introduced in Graham, Pinto and Egel (2016). The function att() allows for sampling weights as well as "clustered standard errors", but these features have not yet been extensively tested. An implementation of the Average Treatment Effect (ATE) estimator introduced in Graham, Pinto and Egel (2012) is planned for a future update. This package is offered "as is", without warranty, implicit or otherwise. While I would appreciate bug reports, suggestions for improvements and so on, I am unable to provide any meaningful user-support. Please e-mail me at bgraham@econ.berkeley.edu Please cite both the code and the underlying source articles listed below when using this code in your research. A simple example script to get started is:: >>>> # Append location of ipt module root directory to systems path >>>> # NOTE: Only required ipt not "permanently" installed >>>> import sys >>>> sys.path.append('/Users/bgraham/Dropbox/Sites/software/ipt/') >>>> # Load ipt package >>>> import ipt as ipt >>>> # View help file >>>> help(ipt.att) >>>> # Read nsw data directly from Rajeev Dehejia's webpage into a >>>> # Pandas dataframe >>>> import numpy as np >>>> import pandas as pd >>>> nsw=pd.read_stata("http://www.nber.org/~rdehejia/data/nsw_dw.dta") >>>> # Make some adjustments to variable definitions in experimental dataframe >>>> nsw['constant'] = 1 # Add constant to observational dataframe >>>> nsw['age'] = nsw['age']/10 # Rescale age to be in decades >>>> nsw['re74'] = nsw['re74']/1000 # Recale earnings to be in thousands >>>> nsw['re75'] = nsw['re75']/1000 # Recale earnings to be in thousands >>>> # Treatment indicator >>>> D = nsw['treat'] >>>> # Balancing moments >>>> t_W = nsw[['constant','black','hispanic','education','age','re74','re75']] >>>> # Propensity score variables >>>> r_W = nsw[['constant']] >>>> # Outcome >>>> Y = nsw['re78'] >>>> # Compute AST estimate of ATT >>>> [gamma_as, vcov_gamma_ast, study_test, auxiliary_test, pi_eff_nsw, \ pi_s_nsw, pi_a_nsw, exitflag] = \ ipt.att(D, Y, r_W, t_W, study_tilt=True) >>>> CODE CITATION --------------- Graham, Bryan S. (2016). "ipt: a Python 2.7 package for causal inference by inverse probability tilting," (Version 0.2.2) [Computer program]. Available at https://github.com/bryangraham/ipt (Accessed 04 May 2016) PAPER CITATIONS --------------- Graham, Bryan S., Cristine Pinto and Daniel Egel. (2012). “Inverse probability tilting for moment condition models with missing data,” Review of Economic Studies 79 (3): 1053 - 1079 Graham, Bryan S., Cristine Pinto and Daniel Egel. (2016). “Efficient estimation of data combination models by the method of auxiliary-to-study tilting (AST),” Journal of Business and Economic Statistics 31 (2): 288 - 301
bryangraham/ipt_py27
Repo for Python 2.7 package implementing program evaluation estimators in Graham, Pinto and Egel (2008, 2012, 2016)
Jupyter NotebookMIT