Skip to content

Repository of causal inference models, with a shared syntax for users to swap between models. This repository also allows users to diagnose their causal estimates absent any ground truth measures.

Notifications You must be signed in to change notification settings

shoepaladin/statanomics

Repository files navigation

statanomics


Here is a one-stop shop of:

  1. causal inference models to estimate average treatment effects (ATE/ATET);
  2. causal inference models to estimate Heterogeneous Treatment Effects (HTE);
  3. diagnostics for assess underlying assumptions needed for causal inference following the Neyman-Rubin's Potential Outcomes model.

The API and syntaxes are centralized, so you can swap one model for another just by changing the functional call!

Installation

available by

pip install --upgrade git+https://github.com/shoepaladin/statanomics#egg=stnomics

causalmodels - ATE

Propensity-score-based models to estimate the average treatment effect and average treatment effect on the treated. While models such as OLS, double robust, and inverse propensity-weighting models are supported.

  • Pending functions:
    • quantile regression
    • regressions with discrete outcomes
    • instrumental variable (IV) regression via 2SLS
    • difference-in-difference
  • No current plans to support:
    • propensity score matching models
    • kernel and local regression models
    • synthetic control models

causalmodels - HTE

Cross sectional methods to estimate heterogeneous treatment effect (HTE) models.

  • Pending support:
    • panel data
    • local linear forests

diagnostics

Various metrics and tests to asses the unconfoundedness and overlap assumptions - following the potential outcomes models. There are no tests for the stable unit treatment value assumption (SUTVA).


(bonus) heterogeneousresiduals

A custom implementation of a heterogeneous treatment effects version for double machine learning, based on The Heterogeneous Residuals model builds on Semenova, Goldman, Chernozhukov, and Taddy (2021).

About

Repository of causal inference models, with a shared syntax for users to swap between models. This repository also allows users to diagnose their causal estimates absent any ground truth measures.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published