statsmodels

Statistical modeling toolkit. OLS, GLM, logistic, ARIMA, time series, hypothesis tests, diagnostics, AIC/BIC, for rigorous statistical inference and econometric analysis.

By davila7 · 434 installs

npx skills add davila7/claude-code-templates --skill statsmodels

Source repository · Upstream listing

Statsmodels: Statistical Modeling and Econometrics Overview Statsmodels is Python's premier library for statistical modeling, providing tools for estimation, inference, and diagnostics across a wide range of statistical methods. Apply this skill for rigorous statistical analysis, from simple linear regression to complex time series models and econometric analyses. When to Use This Skill This skill should be used when: Fitting regression models (OLS, WLS, GLS, quantile regression) Performing generalized linear modeling (logistic, Poisson, Gamma, etc.) Analyzing discrete outcomes (binary, multinomial, count, ordinal) Conducting time series analysis (ARIMA, SARIMAX, VAR, forecasting) Running statistical tests and diagnostics Testing model assumptions (heteroskedasticity, autocorrelation, normality) Detecting outliers and influential observations Comparing models (AIC/BIC, likelihood ratio tests) Estimating causal effects Producing publication ready statistical tables and inference Quick Start Guide Linear Regression (OLS) Logistic Regression (Binary Outcomes) Time Series (ARIMA) Generalized Linear Models (GLM) Core Statistical Modeling Capabilities 1. Linear Regression Models Comprehensive suite of linear models for continuous outcomes with various error structures. Available models: OLS : Standard linear regression with i.i.d. errors WLS : Weighted least squares for heteroskedastic errors GLS : Generalized least squares for arbitrary covariance structure GLSAR : GLS with autoregressive errors for time series Quantile Regression : Conditional quantiles (robust to outliers) Mixed Effects : Hierarchical/multilevel models with random effects Recursive/Rolling : Time varying parameter estimation Key features: Comprehensive diagnostic tests Robust standard errors (HC, HAC, cluster robust) Influence statistics (Cook's distance, leverage, DFFITS) Hypothesis testing (F tests, Wald tests) Model comparison (AIC, BIC, likelihood ratio tests) Prediction with confidence and prediction intervals When to use: Continuous outcome variable, want inference on coefficients, need diagnostics Reference: See references/linear models.md for detailed guidance on model selection, diagnostics, and best practices. 2. Generalized Linear Models (GLM) Flexible framework extending linear models to non normal distributions. Distribution families: Binomial : Binary outcomes or proportions (logistic regression) Poisson : Count data Negative Binomial : Overdispersed counts Gamma : Positive continuous, right skewed data Inverse Gaussian : Positive continuous with specific variance structure Gaussian : Equivalent to OLS Tweedie : Flexible family for semi continuous data Link functions: Logit, Probit, Log, Identity, Inverse, Sqrt, CLogLog, Power Choose based on interpretation needs and model fit Key features: Maximum likelihood estimation via IRLS Deviance and Pearson residuals Goodness of fit statistics Pseudo R squared measures Robust standard errors When to use: Non normal outcomes, need flexible variance and link specifications Reference: See references/glm.md for family selection, link functions, interpretation, and diagnostics. 3. Discrete Choice Models Models for categorical and count outcomes. Binary models: Logit : Logistic regression (odds ratios) Probit : Probit regression (normal distribution) Multinomial models: MNLogit : Unordered categories (3+ levels) Conditional Logit : Choice models with alternative specific variables Ordered Model : Ordinal outcomes (ordered categories) Count models: Poisson : Standard count model Negative Binomial : Overdispersed counts Zero Inflated : Excess zeros (ZIP, ZINB) Hurdle Models : Two stage models for zero heavy data Key features: Maximum likelihood estimation Marginal effects at means or average marginal effects Model comparison via AIC/BIC Predicted probabilities and classification Goodness of fit tests When to use: Binary, categorical, or count outcomes Reference: See references/discrete choice.md for model selection, interpretation, and evaluation. 4. Time Series Analysis Comprehensive time series modeling and forecasting capabilities. Univariate models: AutoReg (AR) : Autoregressive models ARIMA : Autoregressive integrated moving average SARIMAX : Seasonal ARIMA with exogenous variables Exponential Smoothing : Simple, Holt, Holt Winters ETS : Innovations state space models Multivariate models: VAR : Vector autoregression VARMAX : VAR with MA and exogenous variables Dynamic Factor Models : Extract common factors VECM : Vector error correction models (cointegration) Advanced models: State Space : Kalman filtering, custom specifications Regime Switching : Markov switching models ARDL : Autoregressive distributed lag Key features: ACF/PACF analysis for model identification Stationarity tests (ADF, KPSS) Forecasting with prediction intervals Residual diagnostics (Ljung Box, heteroskedasticity) Granger causality testing Impulse response functions (IRF) Forecast error variance decomposition (FEVD) When to use: Time ordered data, forecasting, understanding temporal dynamics Reference: See references/time series.md for model selection, diagnostics, and forecasting methods. 5. Statistical Tests and Diagnostics Extensive testing and diagnostic capabilities for model validation. Residual diagnostics: Autocorrelation tests (Ljung Box, Durbin Watson, Breusch Godfrey) Heteroskedasticity tests (Breusch Pagan, White, ARCH) Normality tests (Jarque Bera, Omnibus, Anderson Darling, Lilliefors) Specification tests (RESET, Harvey Collier) Influence and outliers: Leverage (hat values) Cook's distance DFFITS and DFBETAs Studentized residuals Influence plots Hypothesis testing: t tests (one sample, two sample, paired) Proportion tests Chi square tests Non parametric tests (Mann Whitney, Wilcoxon, Kruskal Wallis) ANOVA (one way, two way, repeated measures) Multiple comparisons: Tukey's HSD Bonferroni correction False Discovery Rate (FDR) Effect sizes and power: Cohen's d, eta squared Power analysis for t tests, proportions Sample size calculations Robust inference: Heteroskedasticity consistent SEs (HC0 HC3) HAC standard errors (Newey West) Cluster robust standard errors When to use: Validating assumptions, detecting problems, ensuring robust inference Reference: See references/stats diagnostics.md for comprehensive testing and diagnostic procedures. Formula API (R style) Statsmodels supports R style formulas for intuitive model specification: Model Selection and Comparison Information Criteria Likelihood Ratio Test (Nested Models) Cross Validation Best Practices Data Preparation 1. Always add constant : Use sm.add constant() unless excluding intercept 2. Check for missing values : Handle or impute before fitting 3. Scale if needed : Improves convergence, interpretation (but not required for tree models) 4. Encode categoricals : Use formula API or manual dummy coding Model Building 1. Start simple : Begin with basic model, add complexity as needed 2. Check assumptions : Test residuals, heteroskedasticity, autocorrelation 3. Use appropriate model : Match model to outcome type (binary→Logit, count→Poisson) 4. Consider alternatives : If assumptions violated, use robust methods or different model Inference 1. Report effect sizes : Not just p values 2. Use robust SEs : When heteroskedasticity or clustering present 3. Multiple comparisons : Correct when testing many hypotheses 4. Confidence intervals : Always report alongside point estimates Model Evaluation 1. Check residuals : Plot residuals vs fitted, Q Q plot 2. Influence diagnostics : Identify and investigate influential observations 3. Out of sample validation : Test on holdout set or cross validate 4. Compare models : Use AIC/BIC for non nested, LR test for nested Reporting 1. Comprehensive summary : Use .summary() for detailed output 2. Document decisions : Note transformations, excluded observations 3. Interpret carefully : Account for link functions (e.g., exp(β) for log link) 4. Visualize : Plot predictions, confidence intervals, diagnostics Common Workflows Workflow 1: Linear Regression Analysis 1. Explore data (plots, descriptives) 2. Fit initial OLS model 3. Check residual diagnostics 4. Test for heteroskedasticity, autocorrelation 5. Check for multicollinearity (VIF) 6. Identify influential observations 7. Refit with robust SEs if needed 8. Interpret coefficients and inference 9. Validate on holdout or via CV Workflow 2: Binary Classification 1. Fit logistic regression (Logit) 2. Check for convergence issues 3. Interpret odds ratios 4. Calculate marginal effects 5. Evaluate classification performance (AUC, confusion matrix) 6. Check for influential observations 7. Compare with alternative models (Probit) 8. Validate predictions on test set Workflow 3: Count Data Analysis 1. Fit Poisson regression 2. Check for overdispersion 3. If overdispersed, fit Negative Binomial 4. Check for excess zeros (consider ZIP/ZINB) 5. Interpret rate ratios 6. Assess goodness of fit 7. Compare models via AIC 8. Validate predictions Workflow 4: Time Series Forecasting 1. Plot series, check for trend/seasonality 2. Test for stationarity (ADF, KPSS) 3. Difference if non stationary 4. Identify p, q from ACF/PACF 5. Fit ARIMA or SARIMAX 6. Check residual diagnostics (Ljung Box) 7. Generate forecasts with confidence intervals 8. Evaluate forecast accuracy on test set Reference Documentation This skill includes comprehensive reference files for detailed guidance: references/linear models.md Detailed coverage of linear regression models including: OLS, WLS, GLS, GLSAR, Quantile Regression Mixed effects models Recursive and rolling regression Comprehensive diagnostics (heteroskedasticity, autocorrelation, multicollinearity) Influence statistics and outlier detection Robust standard errors (HC, HAC, cluster) Hypothesis testing and model comparison references/glm.md Complete guide to generalized linear models: All distribution families (Binomial, Poisson, Gamma, etc.) Link functions and when to use each Model fitting and interpretation Pseudo R squared and goodness of fit Diagnostics and residual analysis Applications (logistic, Poisson, Gamma regression) references/discrete choice.md Comprehensive guide to discrete outcome models: Binary models (Logit, Probit) Multinomial models (MNLogit, Conditional Logit) Count models (Poisson, Negative Binomial, Zero Inflated, Hurdle) Ordinal models Marginal effects and interpretation Model diagnostics and comparison references/time series.md In depth time series analysis guidance: Univariate models (AR, ARIMA, SARIMAX, Exponential Smoothing) Multivariate models (VAR, VARMAX, Dynamic Factor) State space models Stationarity testing and diagnostics Forecasting methods and evaluation Granger causality, IRF, FEVD references/stats diagnostics.md Comprehensive statistical testing and diagnostics: Residual diagnostics (autocorrelation, heteroskedasticity, normality) Influence and outlier detection Hypothesis tests (parametric and non parametric) ANOVA and post hoc tests Multiple comparisons correction Robust covariance matrices Power analysis and effect sizes When to reference: Need detailed parameter explanations Choosing between similar models Troubleshooting convergence or diagnostic issues Understanding specific test statistics Looking for code examples for advanced features Search patterns: Common Pitfalls to Avoid 1. Forgetting constant term : Always us