pymc_models#

Custom PyMC models for causal inference

Classes

BayesianBasisExpansionTimeSeries

Bayesian Structural Time Series Model.

InstrumentalVariableRegression

Custom PyMC model for instrumental linear regression

LinearRegression

Custom PyMC model for linear regression.

PropensityScore

Custom PyMC model for inverse propensity score models

PyMCModel

A wrapper class for PyMC models.

SoftmaxWeightedSumFitter

Weighted sum model with softmax-over-Normal-logits parameterization.

StateSpaceTimeSeries

State-space time series model using pymc-extras.statespace.structural.

SyntheticDifferenceInDifferencesWeightFitter

Bayesian weight fitter for Synthetic Difference-in-Differences.

WeightedSumFitter

Used for synthetic control experiments.