Skip to main content

Diagnostic Plots for Lineare Regression Models. Similar to plot.lm in R.

Project description

Python Library providing Diagnostic Plots for Lineare Regression Models. (Like plot.lm in R.)

I built this, because I missed the diagnostics plots of R for a university project. There are some substitutions in Python for individual charts, but they are spread over different libraries and sometimes don’t show the exact same. My implementation tries to copycat the R-plots, but I didn’t reimplement the R-code: The charts are just based on available documentation.

Installation

Available in PyPi: https://pypi.org/project/lmdiag/

  • Using pip: pip install lmdiag
  • Using pipenv: pipenv install lmdiag

Usage

The plots need a fitted Linear Regression Model created by statsmodels as input.

Fitted Model from Linearmodels <https://bashtage.github.io/linearmodels/doc/index.html> should also work, however, that’s not tested very well.

Example

(See also the more extensive Example Notebook)

import numpy as np
import matplotlib.pyplot as plt
import statsmodels.api as sm
import lmdiag

%matplotlib inline  # In Jupyter

# Generate sample model
np.random.seed(20)
predictor = np.random.normal(size=30, loc=20, scale=3)
response = 5 + 5 * predictor + np.random.normal(size=30)
X = sm.add_constant(predictor)
lm = sm.OLS(response, X).fit()

# Plot chart matrix (and enlarge figure)
plt.figure(figsize=(10,7))
lmdiag.plot(lm);
https://raw.githubusercontent.com/dynobo/lmdiag/master/example.png

Methods

  • Draw matrix of all plots:

    lmdiag.plot(lm)

  • Draw individual plots:

    lmdiag.resid_fit(lm)

    lmdiag.q_q(lm)

    lmdiag.scale_loc(lm)

    lmdiag.resid_lev(lm)

  • Print useful descriptions for interpretations:

    lmdiag.info() (for all plots)

    lmdiag.info('<method name>') (for individual plot)

Development

Disclaimer

This is my very first public python library. Don’t expect everything to work smoothly. I’m happy to receive useful feedback or pull requests.
‘Programming Language :: Python :: 3.6’,

Certification

https://raw.githubusercontent.com/dynobo/lmdiag/master/badge.png

Packaging and Upload to PyPi

  • pipenv run rstcheck README.rst (check syntax)
  • rm -rf ./dist (delete old builds)
  • python setup.py sdist
  • python setup.py bdist_wheel
  • twine upload dist/*
  • Then new release on github…

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Files for lmdiag, version 0.3.7
Filename, size File type Python version Upload date Hashes
Filename, size lmdiag-0.3.7-py3-none-any.whl (7.9 kB) File type Wheel Python version py3 Upload date Hashes View hashes
Filename, size lmdiag-0.3.7.tar.gz (6.8 kB) File type Source Python version None Upload date Hashes View hashes

Supported by

Elastic Elastic Search Pingdom Pingdom Monitoring Google Google BigQuery Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN DigiCert DigiCert EV certificate StatusPage StatusPage Status page