NYC Data Science Academy| Blog
Bootcamps
Lifetime Job Support Available Financing Available
Bootcamps
Data Science with Machine Learning Flagship ๐Ÿ† Data Analytics Bootcamp Artificial Intelligence Bootcamp New Release ๐ŸŽ‰
Free Lesson
Intro to Data Science New Release ๐ŸŽ‰
Find Inspiration
Find Alumni with Similar Background
Job Outlook
Occupational Outlook Graduate Outcomes Must See ๐Ÿ”ฅ
Alumni
Success Stories Testimonials Alumni Directory Alumni Exclusive Study Program
Courses
View Bundled Courses
Financing Available
Bootcamp Prep Popular ๐Ÿ”ฅ Data Science Mastery Data Science Launchpad with Python View AI Courses Generative AI for Everyone New ๐ŸŽ‰ Generative AI for Finance New ๐ŸŽ‰ Generative AI for Marketing New ๐ŸŽ‰
Bundle Up
Learn More and Save More
Combination of data science courses.
View Data Science Courses
Beginner
Introductory Python
Intermediate
Data Science Python: Data Analysis and Visualization Popular ๐Ÿ”ฅ Data Science R: Data Analysis and Visualization
Advanced
Data Science Python: Machine Learning Popular ๐Ÿ”ฅ Data Science R: Machine Learning Designing and Implementing Production MLOps New ๐ŸŽ‰ Natural Language Processing for Production (NLP) New ๐ŸŽ‰
Find Inspiration
Get Course Recommendation Must Try ๐Ÿ’Ž An Ultimate Guide to Become a Data Scientist
For Companies
For Companies
Corporate Offerings Hiring Partners Candidate Portfolio Hire Our Graduates
Students Work
Students Work
All Posts Capstone Data Visualization Machine Learning Python Projects R Projects
Tutorials
About
About
About Us Accreditation Contact Us Join Us FAQ Webinars Subscription An Ultimate Guide to
Become a Data Scientist
    Login
NYC Data Science Acedemy
Bootcamps
Courses
Students Work
About
Bootcamps
Bootcamps
Data Science with Machine Learning Flagship
Data Analytics Bootcamp
Artificial Intelligence Bootcamp New Release ๐ŸŽ‰
Free Lessons
Intro to Data Science New Release ๐ŸŽ‰
Find Inspiration
Find Alumni with Similar Background
Job Outlook
Occupational Outlook
Graduate Outcomes Must See ๐Ÿ”ฅ
Alumni
Success Stories
Testimonials
Alumni Directory
Alumni Exclusive Study Program
Courses
Bundles
financing available
View All Bundles
Bootcamp Prep
Data Science Mastery
Data Science Launchpad with Python NEW!
View AI Courses
Generative AI for Everyone
Generative AI for Finance
Generative AI for Marketing
View Data Science Courses
View All Professional Development Courses
Beginner
Introductory Python
Intermediate
Python: Data Analysis and Visualization
R: Data Analysis and Visualization
Advanced
Python: Machine Learning
R: Machine Learning
Designing and Implementing Production MLOps
Natural Language Processing for Production (NLP)
For Companies
Corporate Offerings
Hiring Partners
Candidate Portfolio
Hire Our Graduates
Students Work
All Posts
Capstone
Data Visualization
Machine Learning
Python Projects
R Projects
About
Accreditation
About Us
Contact Us
Join Us
FAQ
Webinars
Subscription
An Ultimate Guide to Become a Data Scientist
Tutorials
Data Analytics
  • Learn Pandas
  • Learn NumPy
  • Learn SciPy
  • Learn Matplotlib
Machine Learning
  • Boosting
  • Random Forest
  • Linear Regression
  • Decision Tree
  • PCA
Interview by Companies
  • JPMC
  • Google
  • Facebook
Artificial Intelligence
  • Learn Generative AI
  • Learn ChatGPT-3.5
  • Learn ChatGPT-4
  • Learn Google Bard
Coding
  • Learn Python
  • Learn SQL
  • Learn MySQL
  • Learn NoSQL
  • Learn PySpark
  • Learn PyTorch
Interview Questions
  • Python Hard
  • R Easy
  • R Hard
  • SQL Easy
  • SQL Hard
  • Python Easy
Data Science Blog > Machine Learning > Predict Home Prices via Machine Learning in Python

Predict Home Prices via Machine Learning in Python

Paul Choi, Christian Opperman and Melanie Zheng
Posted on Mar 8, 2020
The skills I demoed here can be learned through taking Data Science with Machine Learning bootcamp with NYC Data Science Academy.
Links: Github | Kaggle Challenge | Presentation

Introduction

If you are a real estate investor or potential buyer of a home, how do you determine if the asking price is above or below that homeโ€™s true market value? Most of us would simply predict by looking at other homes with similar features: location, size, and age, the number of bedrooms and bathrooms, etc.

This process is relatively easy and straightforward for one home. But what if youโ€™re looking to purchase many homes, or if youโ€™re wanting to predict prices on homes with very different features? Or perhaps you canโ€™t find a similar house on which to base your comparison. Worse, accurately predicting true value becomes increasingly complex as you add more homes and features. Thankfully, we used machine learning in Python to help solve this particular issue.

The Data

The data we used to develop our machine learning model comes from the Kaggle: Advanced Regression Techniques challenge. This data focuses strictly on residential homes in Ames, Iowa, and was split into two sections - a โ€œtrainโ€ dataset, which included houses with price data attached, and a โ€œtestโ€ dataset, made up of a separate group of houses without pricing information.

The Challenge

We were tasked with predicting the final price of the 1,459 homes in the โ€œtestโ€ dataset by training a machine learning model on the โ€œtrainโ€ dataset. The overall accuracy of our predictions was measured by Kaggle, which had the true prices for the homes in the โ€œtestโ€ dataset. Below, we lay out the framework for how we processed the data, determined our best machine learning model, and how accurate our predictions were when compared to the actual home values.

Our Project Workflow

  1. Exploratory Data Analysis
  2. Data Preprocessing
  3. Modeling & Results
  4. Final Takeaways

Exploratory Data Analysis

As with any data-driven project, it was crucial that we examined the provided data before diving in and beginning to make adjustments, as the results of exploratory analysis necessarily informs how the datasets should be processed before applying machine learning techniques. The โ€œtrainโ€ set contained 1,460 observations with 79 predictor variables (excluding โ€˜SalePriceโ€™ and โ€˜IDโ€™ variables), of which 28 were numeric and 51 were categorical. The โ€œtestโ€ set contained 1,459 observations and the same 79 predictor variables. Below is a summary of the dataโ€™s key characteristics:

Checking the Target Variable: SalePrice

  • After visualizing the histogram and probability plot of the SalePrice variable from the train set, it became clear that home prices are positively (right) skewed and exhibited a non-linear trend (see below). Therefore, we knew that we would need to apply a transformation to this variable to increase the effectiveness of the regression models we intended to employ.

Examining Variable Distribution

  • We also checked for skewness among our input variables, to see if there were any variables that might be unevenly represented within the data. Below, we demonstrate the variables that exhibited the most skewness.

Looking for Missingness

  • Additionally, both the train and test datasets were incomplete, in the sense that many observations had missing values for one or more feature variables. In fact, there were four variables that had over 80% of their data missing, as can be seen from the graph of variables with missing values below. Additionally, we identified that the number of variables with missing values in the test set was higher than that in the train set. Further below, we discuss our methodology to imputing missing data.

Identifying Outliers

  • In order to check for linearity, a condition of linear regression techniques, we looked at the scatterplots of various variables against the target variable SalePrice. When we did so, we identified two outliers in the graph between SalePrice and LivingArea, as seen below. Because these outliers represent a large deviation from the general trend and drastically change the calculation of the linear relationship between these two variables, we knew we needed to address these two points.

Examining Multicollinearity

  • Lastly, we checked for variables with high correlation with each other (multicollinearity). We did so because one of the required assumptions for linear regression models involving multiple variables is that there is little-to-no multicollinearity between feature variables. However, our analysis revealed that a handful of our feature variables could pose a multicollinearity problem, including:
    • First Floor (Sq.Ft.) & Total Basement (Sq.Ft.)
    • Total Rooms Above Ground & General Living Area
    • Garage Area (Sq.Ft.) & Garage Cars (One vs. Two)
    • House Year Built & Garage Year Built

Data Preprocessing

Overall, we implemented the following pipeline for our data processing:

  1. Import raw data
  2. Remove outliers in training data
  3. Impute data:
    • Impute pseudo-missing values
    • Impute true-missing values
    • Re-engineer categorical features as necessary
  4. Add new feature variables
  5. Dummify categorical feature variables
  6. Remove unnecessary feature variables

One important quality of the data that itโ€™s important to reconcile is that there were many more feature variables with missing values in the test dataset than there were in the training dataset. As a result, our group decided on a data processing pipeline that took a function-forward approach, so that the end result would be a single function that would correctly process either the training or the test dataset.

Therefore, the below detailed explanation puts forth our philosophy on first types of variables (pseudo-missing, true-missing, categorical, etc.) and then on specific variables, rather than on an individual dataset.

Outliers

  • An examination of the distribution of our feature space revealed that there were two clear outliers in the relation between SalePrice and GrLivArea (a representation of the overall living area of the house). While the overall relational trend between these two variables is positive and linear, there were two observations with a living area greater than 4000 square feet but a sale price less than $300,000, which had negative effects on the overall trend between the two variables.
  • Because there were only two observations out of the 1,460 observations in our training data, we judged it more efficient to remove those observations than to attempt to transform the GrLivArea variable to reduce the outlier-induced skew.

Data Imputation

Pseudo-Missingness

  • There were a number of feature variables that contained missing values that did not, in fact, represent missing data. Instead of an incomplete observation, missing values in these features represented a lack of the element in question on the property. Our group termed those features with this quality โ€œpseudo-missing.โ€
  • The general philosophy for a given variable X (where X represents a housing feature such as a pool, a fireplace, etc.) was to impute missing values as โ€œNo Xโ€.
  • The fourteen feature variables with pseudo-missingness were: Alley, BsmtCond, BsmtQual, BsmtFinType1, Fence, Fireplace, GarageCond, GarageFinish, GarageQual, GarageType, GarageYrBlt, MasVnrType, MiscFeature, and PoolQC.

Actual Missingness

  • In addition to the feature variables with pseudo-missingness, there were five features where missing values indicated true missing data. Those variables were: Electrical, MasVnrArea, LotFrontage, BsmtExposure, and BsmtFinType2. Because many of our machine learning modesls are unable to handle missing data, it was necessary for us to impute the missing values for these feature variables given the data to which we did have access.
  • By and large, the variables were imputed using Mode imputation. When doing so, feature variables were grouped by other, relevant features (for example, MasVnrArea was grouped by Neighborhood and YearBuilt) to provide a more accurate and granular imputation.
  • Additionally, a handful of variables related to garages and basements exhibited both pseudo-missingness and true missingness. Those variables were first processed to fill any pseudo-missingness. This was achieved by imputed rows that corresponded to missingness in all features related to the relevant housing amenity (garage or basement) with the same methodology described in the pseudo-missingness section above; once that was taken care of, the remaining missing values were imputed using the appropriate mode imputation.

Categorical Variables

  • The distribution of a large number of categorical variables was such that many of the features were dominated by a single value, with the remaining values being more sparsely populated. Because this sparsity resulted in a lack of representative data, and in an attempt to reduce the feature space expansion resulting from the need to dummify categorical features, we consolidated those categorical variables to follow the general pattern of โ€œDominant Class(es), Other.โ€
  • Feature variables engineered in this way were: Exterior1st, RoofMatl, RoofStyle, Condition1, LotShape, Functional, Electrical, Heating, Foundation, and SaleType.
  • Additionally, some categorical variables represented an ordinal ranking - for instance, there were numerous variables related to the quality or condition of some housing feature, which were represented by categorical strings (such as โ€œExcellent,โ€ โ€œFair,โ€ etc.). Because quality and condition are inherently ranked, we converted these variables to ordinal variables, to reduce the need for dummification.

New Feature Variables

  • To reduce the size of the feature space without losing important data, the following feature variables were combined to create new feature variables:
    • 1stFlrSF + 2ndFlrSF + TotalBsmtSF -> TotalSF
    • BsmtFullBath + FullBath -> TotalFullBath
    • BsmtHalfBath + HalfBath -> TotalHalfBath
  • Additionally, new categorical feature variables were created to indicate whether a given home had a garage (IsGarage) or a pool (IsPool).

Dummify Categorical Features

  • All categorical feature variables were dummified using One Hot Encoding for use in linear regression-based machine learning models.

Removing Unnecessary Features

  • Some feature variables were rendered unnecessary by the engineering above. Others were so highly correlated with other variables that including them violated core assumptions of linear models, and yet others had too few observations to be statistically useful.Target Variable: SalePrice
  • Dropped due to no useful information: Id
  • Dropped due to re-engineering: TotalBsmtSF, 1stFlrSF, 2ndFlrSF, BsmtFullBath, BsmtHalfBath, FullBath, HalfBath
  • Dropped for correlation reasons: BsmtFinSF1, BsmtFinSF2, BsmtUnfSF, GrLivArea, GarageYrBlt, GarageCars
  • Dropped for lack of observations: LowQualFinSF, PoolArea

Modeling & Results

After handling all missingness and generating our desired features, we moved on to machine learning-driven modeling. Based on our research, we decided to approach modeling as the following: 

  1. Train various baseline models (or Level 1 models) to predict the SalePrice on the training set.
  2. Predict the SalePrice on the training set using baseline models.
  3. Train a meta model (or a Level 2 model) to predict the SalePrice using the predictions from Step 2.

Modeling Methodology Visually Explained

You may notice that this approach essentially represents a stacked model. On a technical level, we used sklearnโ€™s StackingRegressor to achieve our modeling objective.

Level 1 Models

The various Level 1 Models we generated can be generalized into two categories: linear models and tree based models. Each model was first tuned using sklearnโ€™s GridSearchCV to find the optimal hyperparameters on the train dataset, and then trained on the overall train dataset. Below is a summary of the mean squared errors (our evaluation metric) from the cross validation:

Linear Models:

  • Ridge: Cross Validation Score = 0.11513
  • Lasso: Cross Validation Score = 0.11355
  • Elastic Net: Cross Validation Score = 0.11355

Tree Based Models:

  • CatBoost: Cross Validation Score = 0.11291
  • Gradient Boost: Cross Validation Score = 0.11211
  • LightGBM: Cross Validation Score = 0.11573

Of particular note is that when tuning our Elastic Net model, we found that the optimal rho (the parameter that controls the weighted balance of Ridge and Lasso models in the Elastic Net) was 1, suggesting that the optimal Elastic Net model was essentially the Lasso model. 

After training the Level 1 models, we visualized the modelsโ€™ predicted prices against the actual prices provided by the dataset, as well as the most important feature variables that contributed to each modelโ€™s predictions. As can be seen, although the two different classes of models (linear and tree) had different important features, within each class the models had similar important features.

Additionally, it is important to recognize that the plots for predicted price vs. actual price visually demonstrate that the tree-based models are likely overfitting the training data, a general weakness of tree-based models.

Model 1: Ridge

Model 2: Lasso

Model 3: Elastic Net

Model 4: Catboost

Model 5: Gradient Boost

Model 6: Light GBM

Level 2 Models

For the Level 2 Model, we first tried simply averaging all the predictions from the Level 1 models. We then tried a more complicated model for our meta model. We elected to use a  โ€˜Lassoโ€™ model as our meta model for our second stacking regressor. The table below showcases the cross validated RMSE scores for our Level 1 models as well as both our stacked regressor Level 2 models. Additionally, the table includes the Kaggle scores (the errors between our final predicted prices and the actual prices of the test dataset) for each model.

As can be seen, even the simple average stacked model performed better than any single Level 1 model, with an RMSE score of 0.10921 (the closest Level 1 model is the Gradient Boost model, with an RMSE score of 0.11211). Moreover, our final model, the Lasso stacked regressor, achieved a best cross validation score 0.10895.

Final Takeaways

To conclude, our approach, from data exploration through modeling, resulted in a highly accurate predictive model. Overall, our model placed in the Top 14% of Kaggle submissions and we were happy with our result.

About Authors

Paul Choi

View all posts by Paul Choi >

Christian Opperman

Christian Opperman is a data scientist and former project manager at Mitsui & Co., Ltd. with experience in technical writing, a bachelors in Economics, and a Masters in Writing.
View all posts by Christian Opperman >

Melanie Zheng

Melanie is currently enrolled in Georgia Institute of Technology for Master's Degree in Computer Science with Machine Learning specialization. She previously worked as a product manager at Viacom and project manager at Citigroup.
View all posts by Melanie Zheng >

Leave a Comment

Cancel reply

You must be logged in to post a comment.

No comments found.

View Posts by Categories

All Posts 2399 posts
AI 7 posts
AI Agent 2 posts
AI-based hotel recommendation 1 posts
AIForGood 1 posts
Alumni 60 posts
Animated Maps 1 posts
APIs 41 posts
Artificial Intelligence 2 posts
Artificial Intelligence 2 posts
AWS 13 posts
Banking 1 posts
Big Data 50 posts
Branch Analysis 1 posts
Capstone 206 posts
Career Education 7 posts
CLIP 1 posts
Community 72 posts
Congestion Zone 1 posts
Content Recommendation 1 posts
Cosine SImilarity 1 posts
Data Analysis 5 posts
Data Engineering 1 posts
Data Engineering 3 posts
Data Science 7 posts
Data Science News and Sharing 73 posts
Data Visualization 324 posts
Events 5 posts
Featured 37 posts
Function calling 1 posts
FutureTech 1 posts
Generative AI 5 posts
Hadoop 13 posts
Image Classification 1 posts
Innovation 2 posts
Kmeans Cluster 1 posts
LLM 6 posts
Machine Learning 364 posts
Marketing 1 posts
Meetup 144 posts
MLOPs 1 posts
Model Deployment 1 posts
Nagamas69 1 posts
NLP 1 posts
OpenAI 5 posts
OpenNYC Data 1 posts
pySpark 1 posts
Python 16 posts
Python 458 posts
Python data analysis 4 posts
Python Shiny 2 posts
R 404 posts
R Data Analysis 1 posts
R Shiny 560 posts
R Visualization 445 posts
RAG 1 posts
RoBERTa 1 posts
semantic rearch 2 posts
Spark 17 posts
SQL 1 posts
Streamlit 2 posts
Student Works 1687 posts
Tableau 12 posts
TensorFlow 3 posts
Traffic 1 posts
User Preference Modeling 1 posts
Vector database 2 posts
Web Scraping 483 posts
wukong138 1 posts

Our Recent Popular Posts

AI 4 AI: ChatGPT Unifies My Blog Posts
by Vinod Chugani
Dec 18, 2022
Meet Your Machine Learning Mentors: Kyle Gallatin
by Vivian Zhang
Nov 4, 2020
NICU Admissions and CCHD: Predicting Based on Data Analysis
by Paul Lee, Aron Berke, Bee Kim, Bettina Meier and Ira Villar
Jan 7, 2020

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 ChatGPT 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 football 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 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

NYC Data Science Academy

NYC Data Science Academy teaches data science, trains companies and their employees to better profit from data, excels at big data project consulting, and connects trained Data Scientists to our industry.

NYC Data Science Academy is licensed by New York State Education Department.

Get detailed curriculum information about our
amazing bootcamp!

Please enter a valid email address
Sign up completed. Thank you!

Offerings

  • HOME
  • DATA SCIENCE BOOTCAMP
  • ONLINE DATA SCIENCE BOOTCAMP
  • Professional Development Courses
  • CORPORATE OFFERINGS
  • HIRING PARTNERS
  • About

  • About Us
  • Alumni
  • Blog
  • FAQ
  • Contact Us
  • Refund Policy
  • Join Us
  • SOCIAL MEDIA

    ยฉ 2025 NYC Data Science Academy
    All rights reserved. | Site Map
    Privacy Policy | Terms of Service
    Bootcamp Application