This project delivers a Python package designed to help users determine if Ordinary Least Squares (“OLS”) regression is an appropriate model for their data. The package automates key steps, including performing OLS regression, data formatting checks, assumption validation, and multicollinearity detection, ensuring the data meets the prerequisites for a reliable model. By simplifying these essential tasks and providing clear diagnostics, the package empowers users to confidently assess the feasibility of linear regression and build accurate predictive models.
-
eda_summary: This function uses the training set to check whether the data is formatted correctly to even run a linear regression model and returns a chart that shows the distribution of various features. -
check_multicollinearity: This function detects multicollinearity in the training dataset by computing the variance inflation factor (‘VIF’) and pairwise Pearson Correlation for each numeric feature. -
run_linear_regression: This function performs linear regression with preprocessing using sklearn and outputs evaluation scoring metrics. -
qq_and_residuals_plot: This function tests the normality of residuals and homoscedasticity assumptions for a linear regression model by creating the Q-Q and Residuals vs Fitted Values plots, respectively.
linreg_ally provides an alternative to existing packages like StatAssume, which automates model fitting, assumptions checking, and dashboard generation in a single function, and lrasm, which provides specific functions that test certain key assumptions in OLS regression. linreg_ally differentiates itself by providing intuitive, user-friendly functions that guide users through every stage of training an OLS regression model – starting from explanatory data analysis and data preprocessing, progressing to model fitting and evaluating key assumptions in OLS regression.
$ pip install linreg_allyTo use this package:
- Install the package via poetry, run the following command from the root directory of this repo. If poetry hasn't been set up yet, please following this link for installtion.
poetry install- Import the required functions from the package
import linreg_ally
from linreg_ally.eda import eda_summary
from linreg_ally.multicollinearity import check_multicollinearity
from linreg_ally.models import run_linear_regression
from linreg_ally.plotting import qq_and_residuals_plot- For individual function usage, visit our read the docs page to see individual function usage.
To run the tests, run the following commands from the root directory of this repo:
pytest tests/Interested in contributing? Check out the contributing guidelines. Please note that this project is released with a Code of Conduct. By contributing to this project, you agree to abide by its terms.
linreg_ally was created by Paramveer Singh, Merari Santana-Carbajal, Cheng Zhang, and Alex Wong. It is licensed under the terms of the MIT license.
linreg_ally was created with cookiecutter and the py-pkgs-cookiecutter template.