Studying Data to Predict Home Prices in Ames

The skills the author demoed here can be learned through taking Data Science with Machine Learning bootcamp with NYC Data Science Academy.


Data shows purchasing a house is generally a task that requires research and compromises to find one that fits the right budget and is built with certain desired features. As a team, we entered the House Prices: Advanced Regression Techniques Kaggle competition to exercise our machine learning skills. The competition entailed predicting housing prices in Ames, Iowa.

The training dataset included 1460 houses (observations), 79 features, and the Sale Price of each house. The test set included 1459 houses, the same 79 features, but no Sale Prices.

In this post, we detail our process as we work through the data with Python, constructing various regression models with the Sklearn package with the goal of optimizing our prediction score.

Please see here to view our code and shared Github repository.

Our  model-building workflow:

  1. Exploratory Data Analysis (EDA)
  2. Data Preparation
  3. Feature Importance
  4. Modeling & Results


Exploratory Data Analysis & Data Preparation

To build a successful model, one must first understand the data. We evaluated all features one by one to determine which would need to be manipulated in our data preparation step.   

Sale Price (Response Variable)

We first looked at our response variable, Sale Price, which appeared to have a right-skewed distribution. Given that the data is continuous and linear models work under the assumption of normality, we applied a log + 1 transformation to normalize the distribution.

Studying Data to Predict Home Prices in Ames




There were 34 features total that had missing values in the training and test set combined. Due to the fact that missing data could be random, skewed toward one level of a variable, or signify a value, we were careful to evaluate each feature.Studying Data to Predict Home Prices in Ames

Most commonly, missingness represented a zero value, the lack of the feature at a residence. This was true for Fireplace Quality, Basement Quality, Garage Condition, Pool Quality, etc. These were usually filled with a value such as “NoFireplace”, “NoBasement”, etc.

In cases where there was a very low percentage of missingness, we evaluated according to the rest of the data and outside context. For example for the “Electrical” feature, there was only one value missing and the vast majority of houses were categorized under standard circuit breaker, so the missing value was categorized under the majority as well.

For Lot Frontage we used k-nearest-neighbors (kNN) to fill in the missing values given that we could not assume that Lot Frontage was something that did not exist. The kNN method would also be taking into account an average of all the features. We conducted a grid search to find the best value for neighbors (k) by looping through and evaluating the negative mean squared error (-MSE) at each level of k.  After evaluating we tuned the model to k = 14.

Studying Data to Predict Home Prices in Ames


Categorical Variables

There were two types of categorical variables in the data, ordinal and nominal categorical. Ordinal categorical variables are those that have an inherent order. The most obvious example of these variables are those with classes measuring quality, such as “Poor”, “Fair”, “Average/Typical”, “Good”, “Excellent”. We assigned these ordinal categorical variables a numerical value based on their corresponding rank. Nominal categorical variables, in contrast, have no inherent order.  Most of these values were dummified so that a new variable was created for each class within a variable with “1” value if the class was present and “0” value if the class was not.

While performing EDA we identified categorical variables to find those with a clear relationship to the output variable, Sale Price, and then used target encoding to change the category to ordinal.  One example of this type of variable was MSZoning. In the boxplot below showing MSZoning vs Sale Price, it’s clear that there is an order from FV (Floating Village Residential) to C (Commercial). We therefore assigned numbers to the ranking found in the below boxplot and also combined RH (Residential High Density), and RM (Residential Medium Density) into one since these were similar.


Studying Data to Predict Home Prices in Ames



When looking at the relationship between GrLivArea (Above Ground Living Area) and SalePrice, we observed four outliers that were distant and not in line with the other observations. All four outliers were greater than 4,000 square feet in above ground living area. The two that we ended up removing also had an abnormally low sale price. These were removed as they did not follow the general linear trend of the data, and removing these points helped to improve our prediction score.

Studying Data to Predict Home Prices in Ames



Given that normally distributed data is a key assumption underlying many machine learning algorithms, we evaluated the skew of our our input variables alongside our output variables using the skew function in the Scipy package. We determined a threshold of 0.75 and applied a log +1 transformation to all features above the determined threshold to reduce skew.

Ex. LotFrontage


Multicollinearity in the data needs to be addressed given that multiple regression models assume that independent variables are not highly correlated with each other. We approximated for this through building a correlation matrix.

We identified 17 pairs of features with a correlation greater than a determined threshold of 0.65 to each other, and having a lower correlation to the response variable, SalePrice. Then, we removed the feature of each pair that was less correlated to SalePrice. We also removed 4 variables with very low correlation to SalePrice, regardless of other correlations.

Overall, we removed 21 features from the dataset.


Feature Engineering

We created two additional features for our dataset based on contextual understanding.

  1. ‘TotalBsmtSF’, ‘1stFlrSF’, and ‘2ndFlrSF’ were aggregated into a single ‘TotalSF’ variable
  2. We created a ‘HouseAge’ variable, subtracting the year built for each home from the most recent one.


Feature Importance

Several modelling methods quantitatively rank features, an output which is very useful in providing a clear and interpretable understanding of which features in our dataset are the most important.

We examined feature importance via two methods, Lasso Regression and Random Forest, while also trying out Principal Component Analysis (PCA) to reduce the dimensionality of all features used.


Lasso Regression

Lasso Regression applies a regularization term to the modeling algorithm which reduces the size of the coefficients in a regression equation, eventually reducing to 0 the features deemed least important to the algorithm.

We optimized the regularization parameter alpha of our lasso regression through a grid search via the LassoCV function in the scikit learn package. The absolute coefficients of variables in its output represent the most important features.




To compare we also ran the RandomForestRegressor within the scikit learn package to see if similar variables would pop as the most important. We used the feature importance measure from the RandomForestRegressor to obtain the ranking.

In both the Lasso and RandomForest outputs, we saw that OverallQual and TotalSF popped as the top two most important features.


Principal Component Analysis

Lastly, we tested applying Principal Component Analysis (PCA) to our dataset for dimensionality reduction, but did not apply the results in the end given that the independent variables after applying were not interpretable.

PCA works by rotating the axes so that features are reduced to the best combinations together. Below is a plot of our results:


There is a clear elbow, a flattening of the negative slope, at approximately 10 components, demonstrating that via PCA, the majority of the variance of the dataset can be condensed into 10 features.


Model Selection

We used 10 fold cross validation on the training set and tested several different models and combinations. Initially, we tested both linear and non-linear single models. For linear models, we tested Lasso, Ridge, and Multiple Linear Regression, while the non-linear models tested were Support Vector Machines, Random Forest, and XGBoost. The hyperparameters for each model and their cross validation scores are outlined below.

For single model selection, we obtained the best results with the Support Vector Regressor using a radial kernel.

Ensembling and Stacking

To improve results, we also tested ensembling and stacking methods. Ensembling entails averaging predicted y values from different models. Stacking uses machine learning models to perform regression on predicted y values from different models. The below chart outlines the top five stacking or ensembling submissions as well as hyperparameters that were used and the score we received from Kaggle.

Our top model was a stacked model using Lasso, SVR (rbf), XGBoost, Random Forest, Neural Network Perception, with a Lasso Meta-Regressor. The chart below visualizes our best model’s predicted price vs. actual on our training set, since the test set did not contain the Sales Price data.

While our best model fit relatively closely for most homes, as sales price increased beyond $350,000, the error increased as the model consistently under-predicted the sales price.


Conclusions/Next Steps

  • One very important thing we learnt through this project is how crucial is data cleaning and preprocessing for machine learning models to perform well
  • Although our end model produced a relatively low RMSLE of .12020, we lost some of the interpretability of the model. This is inherent of any complex model (trade off between accuracy and simplicity)
  • We can try to further refine our model by  experimenting more with hyperparameter tuning
  • We can complement the dataset with other demographic information, such as crime rate, literacy, school district etc., from other publicly available sources


About Authors

Franklin Dickinson

Business Strategist. Creative Technologist. Critical Thinker. Current fellow at NYC Data Science Academy, utilizing R, Python, machine learning, and big data to test intuitions and drive insights.
View all posts by Franklin Dickinson >

Leave a Comment

No comments found.

View Posts by Categories

Our Recent Popular Posts

View Posts by Tags

#python #trainwithnycdsa 2019 2020 Revenue 3-points agriculture air quality airbnb airline alcohol Alex Baransky algorithm alumni Alumni Interview Alumni Reviews Alumni Spotlight alumni story Alumnus ames dataset ames housing dataset apartment rent API Application artist aws bank loans beautiful soup Best Bootcamp Best Data Science 2019 Best Data Science Bootcamp Best Data Science Bootcamp 2020 Best Ranked Big Data Book Launch Book-Signing bootcamp Bootcamp Alumni Bootcamp Prep boston safety Bundles cake recipe California Cancer Research capstone car price Career Career Day citibike classic cars classpass clustering Coding Course Demo Course Report covid 19 credit credit card crime frequency crops D3.js data data analysis Data Analyst data analytics data for tripadvisor reviews data science Data Science Academy Data Science Bootcamp Data science jobs Data Science Reviews Data Scientist Data Scientist Jobs data visualization database Deep Learning Demo Day Discount disney dplyr drug data e-commerce economy employee employee burnout employer networking environment feature engineering Finance Financial Data Science fitness studio Flask flight delay gbm Get Hired ggplot2 googleVis H20 Hadoop hallmark holiday movie happiness healthcare frauds higgs boson Hiring hiring partner events Hiring Partners hotels housing housing data housing predictions housing price hy-vee Income Industry Experts Injuries Instructor Blog Instructor Interview insurance italki Job Job Placement Jobs Jon Krohn JP Morgan Chase Kaggle Kickstarter las vegas airport lasso regression Lead Data Scienctist Lead Data Scientist leaflet league linear regression Logistic Regression machine learning Maps market matplotlib Medical Research Meet the team meetup methal health miami beach movie music Napoli NBA netflix Networking neural network Neural networks New Courses NHL nlp NYC NYC Data Science nyc data science academy NYC Open Data nyc property NYCDSA NYCDSA Alumni Online Online Bootcamp Online Training Open Data painter pandas Part-time performance phoenix pollutants Portfolio Development precision measurement prediction Prework Programming public safety PwC python Python Data Analysis python machine learning python scrapy python web scraping python webscraping Python Workshop R R Data Analysis R language R Programming R Shiny r studio R Visualization R Workshop R-bloggers random forest Ranking recommendation recommendation system regression Remote remote data science bootcamp Scrapy scrapy visualization seaborn seafood type Selenium sentiment analysis sentiment classification Shiny Shiny Dashboard Spark Special Special Summer Sports statistics streaming Student Interview Student Showcase SVM Switchup Tableau teachers team team performance TensorFlow Testimonial tf-idf Top Data Science Bootcamp Top manufacturing companies Transfers tweets twitter videos visualization wallstreet wallstreetbets web scraping Weekend Course What to expect whiskey whiskeyadvocate wildfire word cloud word2vec XGBoost yelp youtube trending ZORI