Update: The current implementation still not finialize the optimization process when using
unbiased_gradient
as in theexperiments/demand_simulation_mnist.py
. You can still use the code inexperiments/demand_simulation.py
. See Section 3.1 in Hartford17 for more expaination about the optimization for DeepIV.
An implementation of "deep instrument variable methods" by the LATEST tensorflow and keras
tensorflow==2.3.0
keras==2.3.1
python==3.7 or 3.8
- Run:
pip install git+https://github.com/XingruiWang/DeepIV.git
A package for counterfactual prediction using deep instrument variable methods that builds on Keras_.
You can read how the method works in our DeepIV_ paper.
If you use this package in your research, please cite it as::
@inproceedings{Hartford17,
author = {Jason Hartford and
Greg Lewis and
Kevin Leyton-Brown and
Matt Taddy},
title = {Deep IV: A Flexible Approach for Counterfactual Prediction},
booktitle = {Proceedings of the 34th International Conference on Machine Learning,
{ICML} 2017, Sydney, Australia, 6-11 August 2017},
pages = {1--9},
year = {2017}
}
- Free software: MIT license
- Documentation: https://deepiv.readthedocs.io.
To use DeepIV, you can simply naviage to to the DeepIV directory on your machine and run:
pip install .
You can then use the package by simply running: import deepiv in python. See the examples directory for example usage.
The package is currently under active development, so you may want to install it using the following command:
pip install -e .
By doing this, every time you git pull an update, it will be reflected in your installation.
The DeepIV package is simply a subclass of the Keras Model class that provides the necessary functions for fitting Deep instrumental variable models. Because of this, you can think of it as a drop-in replacement of the Keras Model object. The DeepIV procedure consists of two stages:
- Fit the Treatment model.
- Fit the Response model that takes the fitted Treatment model as input.
Example usage is shown in the experiments directory.
demand_simulation.py
gives a simple example using a feedforward network for both the treatment and the response models.
demand_simulation_mnist.py
is a little more complicated: it uses a convolutation network to fit an image embedding and then concatinates the embedding with other features to fit the network.
Both those examples use simulated data where ground truth is known, so they can report the causal mean squared error. On real data this isn't possible, so we advise that you use a holdout set to tune hyperparameters of the network (or cross validation in the case of small networks). You can choose hyperparameters based on the losses returned at each stage (see the paper for details on why this works).
DeepIV should be compatable with all Keras layers, so the Keras_ documentation is a good place to learn about designing network architectures. Feel free to file a bug report if something doesn't work.
This package was created with Cookiecutter_ and the audreyr/cookiecutter-pypackage
_ project template.
.. _DeepIV: http://proceedings.mlr.press/v70/hartford17a.html
.. _Keras: https://keras.io
.. _Cookiecutter: https://github.com/audreyr/cookiecutter
.. _audreyr/cookiecutter-pypackage
: https://github.com/audreyr/cookiecutter-pypackage