This is a Python implementation of the Instrumtented Principal Components Analysis framework by Kelly, Pruitt, Su (2017).
Exemplary use of the ipca package. The data is the seminal Grunfeld data set as provided on statsmodels. Note, the fit method takes a panel of data, X, with the following columns:
- entity id (numeric)
- time (numeric)
- and following columns contain characteristics.
as well as a series of dependent variables, y, of the same length as X.
import numpy as np from statsmodels.datasets import grunfeld data = grunfeld.load_pandas().data data.year = data.year.astype(np.int64) # Establish unique IDs to conform with package N = len(np.unique(data.firm)) ID = dict(zip(np.unique(data.firm).tolist(),np.arange(1,N+1))) data.firm = data.firm.apply(lambda x: ID[x]) # use multi-index for panel groups data = data.set_index(['firm', 'year']) y = data['invest'] X = data.drop('invest', axis=1) # Call ipca from ipca import InstrumentedPCA regr = InstrumentedPCA(n_factors=1, intercept=False) regr = regr.fit(X=X, y=y) Gamma, Factors = regr.get_factors(label_ind=True)The latest release can be installed using pip
pip install ipcaThe master branch can be installed by cloning the repo and running setup
git clone https://github.com/bkelly-lab/ipca.git cd ipca python setup.py installThe lastest documenation is published HERE.
With the exception of Python 3.6+, which is a hard requirement, the others are the version that are being used in the test environment. It is possible that older versions work.
- Python 3.7+:
- NumPy (1.19+)
- SciPy (1.6+)
- Numba (0.53+)
- progressbar (2.5+)
- joblib (1.0.1+)
- pandas (1.2.3+)
- scikit-learn (0.24+)
- pytest (4.3+)
- statsmodels (0.11+)
The implementation is inspired by the MATLAB code for IPCA made available on Seth Pruitt's website.
- Kelly, Pruitt, Su (2017). "Instrumented Principal Components Analysis" SSRN
The package is still in the development phase, hence please share your comments and suggestions with us.
Contributions welcome!
- Matthias Buechner, Leland Bybee