Regression Models Python Packages

statsmodels

Statsmodels: statistical modeling and econometrics in Python

bambi

BAyesian Model-Building Interface (Bambi) in Python.

orbit-ml

A Python package for Bayesian forecasting with object-oriented design and probabilistic models under the hood.

stargazer

Python implementation of the R stargazer multiple regression model creation tool

greybox

Regression model building and forecasting in R

myfm

A Python/C++ implementation of Bayesian Factorization Machines

dominance-analysis

This package can be used for dominance analysis or Shapley Value Regression for finding relative importance of predictors on given dataset. This library can be used for key driver analysis or marginal resource allocation models.

trust-free

TRUST - Decision Trees with Sparse Elastic Net leaves, Random Forest accuracy and automated explanations

statstests

Statstests: a Python package that provides a complement of process and statistical tests for statsmodels statistical models.

pyprocessmacro

A Python library for moderation, mediation and conditional process analysis.

treeig

Exact Integrated Gradients for tree ensembles, including XGBoost, LightGBM, Random Forests, and Gradient Boosting. Computes fast, exact feature attributions for tree models without numerical gradients, numerical integration, Monte Carlo sampling, or path approximations.

edef

Feature attribution for realized model fit (rather than predictions). EDEF provides exact additive decomposition of predictive performance for linear, PyTorch, JAX, and tree-based models, including standard errors, grouped inference, and exact TreeIG decompositions for tree-based models.

lazypredict-nightly

Lazy Predict Nightly enables rapid benchmarking of machine learning models with just a few lines of code. It helps identify the best-performing algorithms on your dataset without requiring hyperparameter tuning.

bess

Best Subset Selection algorithm for Regression, Classification, Count, Survival analysis

appelpy

Applied Econometrics Library for Python

galpynostatic

:zap::battery: A Python/C++ package with physics-based and data-driven models to predict optimal conditions for fast-charging lithium-ion batteries

midlearn

A scikit-learn compatible Python wrapper of the midr R package

hlr

HLR - Hierarchical Linear Regression for Python

sparsereg

Modern sparse linear regression

peak-engines

Machine learning framework built on second-order optimization

choquet-classifier-glenscalai

Implementation of the Choquet classifier using the scikit-learn API design.

intelelm

IntelELM: A Python Framework for Intelligent Metaheuristic-based Extreme Learning Machine

metaperceptron

MetaPerceptron: A Standardized Framework For Metaheuristic-Driven Multi-layer Perceptron Optimization

genetest

Python package for efficient genetic association analyses