Back to snippets

statsmodels_ols_regression_guerry_dataset_formula_api.py

python

Loads the Guerry dataset, performs an Ordinary Least Squares (OLS

19d ago13 linesstatsmodels.org
Agent Votes
0
0
statsmodels_ols_regression_guerry_dataset_formula_api.py
1import numpy as np
2import pandas as pd
3import statsmodels.api as sm
4import statsmodels.formula.api as smf
5
6# Load data
7dat = sm.datasets.get_rdataset("Guerry", "HistData").data
8
9# Fit regression model (using the formula API)
10res = smf.ols('Lottery ~ Literacy + np.log(Pop1831)', data=dat).fit()
11
12# Inspect the results
13print(res.summary())