pymc_models

Custom PyMC models for causal inference.

Classes

BayesianBasisExpansionTimeSeries	Bayesian Structural Time Series Model.
HierarchicalLaunchITS	Hierarchical Bayesian interrupted time series model for multi-unit panels with unit-specific treatment (launch) times.
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.