Predicting Housing Prices with Linear Regression using Python, pandas, and statsmodels
In this post, we'll walk through building linear regression models to predict housing prices resulting from economic activity.
LearnDataSci is reader-supported. When you purchase through links on our site, earned commissions help support our team of writers, researchers, and designers at no extra cost to you.
You should already know:
- Python fundamentals
- Some Pandas experience
Learn both interactively through dataquest.io
This post will walk you through building linear regression models to predict housing prices resulting from economic activity.
Future posts will cover related topics such as exploratory analysis, regression diagnostics, and advanced regression modeling, but I wanted to jump right in so readers could get their hands dirty with data. If you would like to see anything in particular, feel free to leave a comment below.
Let's dive in.
Article Resources
- Notebook and Data: GitHub
- Libraries: numpy, pandas, matplotlib, seaborn, statsmodels
What is Regression?
Linear regression is a model that predicts a relationship of direct proportionality between the dependent variable (plotted on the vertical or Y axis) and the predictor variables (plotted on the X axis) that produces a straight line, like so:
Linear regression will be discussed in greater detail as we move through the modeling process.
Variable Selection
For our dependent variable we'll use housing_price_index
(HPI), which measures price changes of residential housing.
For our predictor variables, we use our intuition to select drivers of macro- (or “big picture”) economic activity, such as unemployment, interest rates, and gross domestic product (total productivity). For an explanation of our variables, including assumptions about how they impact housing prices, and all the sources of data used in this post, see here.
Reading in the Data with pandas
Before anything, let's get our imports for this tutorial out of the way.
The first import is just to change how tables appear in the accompanying notebook, the rest will be explained once they're used:
You can grab the data using the pandas read_csv
method directly from GitHub. Alternatively, you can download it locally.
Once we have the data, invoke pandas' merge
method to join the data together in a single dataframe for analysis. Some data is reported monthly, others are reported quarterly. No worries. We merge the dataframes on a certain column so each row is in its logical place for measurement purposes. In this example, the best column to merge on is the date column. See below.
Let's get a quick look at our variables with pandas' head
method. The headers in bold text represent the date and the variables we'll test for our model. Each row represents a different time period.
date | sp500 | consumer_price_index | long_interest_rate | housing_price_index | total_unemployed | more_than_15_weeks | not_in_labor_searched_for_work | multi_jobs | leavers | losers | federal_funds_rate | total_expenditures | labor_force_pr | producer_price_index | gross_domestic_product | |
---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
0 | 2011-01-01 | 1282.62 | 220.22 | 3.39 | 181.35 | 16.2 | 8393 | 2800 | 6816 | 6.5 | 60.1 | 0.17 | 5766.7 | 64.2 | 192.7 | 14881.3 |
1 | 2011-04-01 | 1331.51 | 224.91 | 3.46 | 180.80 | 16.1 | 8016 | 2466 | 6823 | 6.8 | 59.4 | 0.10 | 5870.8 | 64.2 | 203.1 | 14989.6 |
2 | 2011-07-01 | 1325.19 | 225.92 | 3.00 | 184.25 | 15.9 | 8177 | 2785 | 6850 | 6.8 | 59.2 | 0.07 | 5802.6 | 64.0 | 204.6 | 15021.1 |
3 | 2011-10-01 | 1207.22 | 226.42 | 2.15 | 181.51 | 15.8 | 7802 | 2555 | 6917 | 8.0 | 57.9 | 0.07 | 5812.9 | 64.1 | 201.1 | 15190.3 |
4 | 2012-01-01 | 1300.58 | 226.66 | 1.97 | 179.13 | 15.2 | 7433 | 2809 | 7022 | 7.4 | 57.1 | 0.08 | 5765.7 | 63.7 | 200.7 | 15291.0 |
Usually, the next step after gathering data would be exploratory analysis. Exploratory analysis is the part of the process where we analyze the variables (with plots and descriptive statistics) and figure out the best predictors of our dependent variable.
For the sake of brevity, we'll skip the exploratory analysis. Keep in the back of your mind, though, that it's of utmost importance and that skipping it in the real world would preclude ever getting to the predictive section.
We'll use ordinary least squares (OLS), a basic yet powerful way to assess our model.
Ordinary Least Squares Assumptions
OLS measures the accuracy of a linear regression model.
OLS is built on assumptions which, if held, indicate the model may be the correct lens through which to interpret our data. If the assumptions don't hold, our model's conclusions lose their validity.
Take extra effort to choose the right model to avoid Auto-esotericism/Rube-Goldberg’s Disease.
Here are the OLS assumptions:
- Linearity: A linear relationship exists between the dependent and predictor variables. If no linear relationship exists, linear regression isn't the correct model to explain our data.
- No multicollinearity: Predictor variables are not collinear, i.e., they aren't highly correlated. If the predictors are highly correlated, try removing one or more of them. Since additional predictors are supplying redundant information, removing them shouldn't drastically reduce the Adj. R-squared (see below).
- Zero conditional mean: The average of the distances (or residuals) between the observations and the trend line is zero. Some will be positive, others negative, but they won't be biased toward a set of values.
- Homoskedasticity: The certainty (or uncertainty) of our dependent variable is equal across all values of a predictor variable; that is, there is no pattern in the residuals. In statistical jargon, the variance is constant.
- No autocorrelation (serial correlation): Autocorrelation is when a variable is correlated with itself across observations. For example, a stock price might be serially correlated if one day's stock price impacts the next day's stock price.
Let's begin modeling.
Simple Linear Regression
Simple linear regression uses a single predictor variable to explain a dependent variable. A simple linear regression equation is as follows:
$$y_i = \alpha + \beta x_i + \epsilon_i$$
Where:
$y$ = dependent variable
$\beta$ = regression coefficient
$\alpha$ = intercept (expected mean value of housing prices when our independent variable is zero)
$x$ = predictor (or independent) variable used to predict Y
$\epsilon$ = the error term, which accounts for the randomness that our model can't explain.
Using statsmodels' ols
function, we construct our model setting housing_price_index
as a function of total_unemployed
. We assume that an increase in the total number of unemployed people will have downward pressure on housing prices. Maybe we're wrong, but we have to start somewhere!
The code below shows how to set up a simple linear regression model with total_unemployment
as our predictor variable.
To explain:
Adj. R-squared indicates that 95% of housing prices can be explained by our predictor variable, total_unemployed
.
The regression coefficient (coef) represents the change in the dependent variable resulting from a one unit change in the predictor variable, all other variables being held constant. In our model, a one unit increase in total_unemployed
reduces housing_price_index
by 8.33. In line with our assumptions, an increase in unemployment appears to reduce housing prices.
The standard error measures the accuracy of total_unemployed
's coefficient by estimating the variation of the coefficient if the same test were run on a different sample of our population. Our standard error, 0.41, is low and therefore appears accurate.
The p-value means the probability of an 8.33 decrease in housing_price_index
due to a one unit increase in total_unemployed
is 0%, assuming there is no relationship between the two variables. A low p-value indicates that the results are statistically significant, that is in general the p-value is less than 0.05.
The confidence interval is a range within which our coefficient is likely to fall. We can be 95% confident that total_unemployed
's coefficient will be within our confidence interval, [-9.185, -7.480].
Regression Plots
Please see the four graphs below.
- The “Y and Fitted vs. X” graph plots the dependent variable against our predicted values with a confidence interval. The inverse relationship in our graph indicates that
housing_price_index
andtotal_unemployed
are negatively correlated, i.e., when one variable increases the other decreases. - The “Residuals versus
total_unemployed
” graph shows our model's errors versus the specified predictor variable. Each dot is an observed value; the line represents the mean of those observed values. Since there's no pattern in the distance between the dots and the mean value, the OLS assumption of homoskedasticity holds. - The “Partial regression plot” shows the relationship between
housing_price_index
andtotal_unemployed
, taking in to account the impact of adding other independent variables on our existingtotal_unemployed
coefficient. We'll see later how this same graph changes when we add more variables. - The Component and Component Plus Residual (CCPR) plot is an extension of the partial regression plot, but shows where our trend line would lie after adding the impact of adding our other independent variables on our existing
total_unemployed
coefficient. More on this plot here.
The next plot graphs our trend line (green), the observations (dots), and our confidence interval (red).
Multiple Linear Regression
Mathematically, multiple linear regression is:
$$Y = \beta_0 + \beta_1 x_1 + \beta_2 x_2 + ... + \beta_k x_k + \epsilon$$
We know that unemployment cannot entirely explain housing prices. To get a clearer picture of what influences housing prices, we add and test different variables and analyze the regression results to see which combinations of predictor variables satisfy OLS assumptions, while remaining intuitively appealing from an economic perspective.
We arrive at a model that contains the following variables: fed_funds
, consumer_price_index
, long_interest_rate
, and gross_domestic_product
, in addition to our original predictor, total_unemployed
.
Adding the new variables decreased the impact of total_unemployed
on housing_price_index
. total_unemployed
‘s impact is now more unpredictable (standard error increased from 0.41 to 2.399), and, since the p-value is higher (from 0 to 0.943), less likely to influence housing prices.
Although total_unemployed
may be correlated with housing_price_index
, our other predictors seem to capture more of the variation in housing prices. The real-world interconnectivity among our variables can't be encapsulated by a simple linear regression alone; a more robust model is required. This is why our multiple linear regression model's results change drastically when introducing new variables.
That all our newly introduced variables are statistically significant at the 5% threshold, and that our coefficients follow our assumptions, indicates that our multiple linear regression model is better than our simple linear model.
The code below sets up a multiple linear regression with our new predictor variables.
Another Look at Partial Regression Plots
Now let's plot our partial regression graphs again to visualize how the total_unemployed
variable was impacted by including the other predictors. The lack of trend in the partial regression plot for total_unemployed
(in the figure below, upper right corner), relative to the regression plot for total_unemployed (above, lower left corner), indicates that total unemployment isn't as explanatory as the first model suggested. We also see that the observations from the latest variables are consistently closer to the trend line than the observations for total_unemployment
, which reaffirms that fed_funds
, consumer_price_index
, long_interest_rate
, and gross_domestic_product
do a better job of explaining housing_price_index
.
These partial regression plots reaffirm the superiority of our multiple linear regression model over our simple linear regression model.
Conclusion
We have walked through setting up basic simple linear and multiple linear regression models to predict housing prices resulting from macroeconomic forces and how to assess the quality of a linear regression model on a basic level.
To be sure, explaining housing prices is a difficult problem. There are many more predictor variables that could be used. And causality could run the other way; that is, housing prices could be driving our macroeconomic variables; and even more complex still, these variables could be influencing each other simultaneously.
I encourage you to dig into the data and tweak this model by adding and removing variables while remembering the importance of OLS assumptions and the regression results.
Most importantly, know that the modeling process, being based in science, is as follows: test, analyze, fail, and test some more.
Navigating Pitfalls
This post is an introduction to basic regression modeling, but experienced data scientists will find several flaws in our method and model, including:
- No Lit Review: While it's tempting to dive in to the modeling process, ignoring the existing body of knowledge is perilous. A lit review might have revealed that linear regression isn't the proper model to predict housing prices. It also might have improved variable selection. And spending time on a lit review at the outset can save a lot of time in the long run.
- Small sample size: Modeling something as complex as the housing market requires more than six years of data. Our small sample size is biased toward the events after the housing crisis and is not representative of long-term trends in the housing market.
- Multicollinearity: A careful observer would've noticed the warnings produced by our model regarding multicollinearity. We have two or more variables telling roughly the same story, overstating the value of each of the predictors.
- Autocorrelation: Autocorrelation occurs when past values of a predictor influence its current and future values. Careful reading of the Durbin-Watson score would've revealed that autocorrelation is present in our model.
In a future post, we'll attempt to resolve these flaws to better understand the economic predictors of housing prices.