House Prices Prediction Using Machine Learning


The goal of this project was to utilize supervised machine learning techniques to predict the price of house located in Ames, Iowa. The dataset was provided by Kaggle, an online community of data scientists and machine learners, owned by Google.

This dataset provides 1,460 observations and 80 features, including multiple aspects of the house that may or may not help predict the fluctuation of the house prices.

The strategic approach our team adopted was to derive meaning from the dataset through numerical and visual exploratory data analysis, drop statistically irrelevant features and features with high multicollinearity, impute missing values, create new features, and then apply different supervised machine-learning algorithms to predict the housing sales prices. 


As you can see below, transforming the data was important due to the skewness of the overall data.

Exhibit 1. Skewness of Features

The first transformation we performed was on the sale price. You can see that before the log transformation, the data was heavily right-skewed. After the log transformation, though, the sales price achieved an approximately normal distribution, and the skewness value dropped from 0.188 to 0.12

Exhibit 2. Density Plots of Target Variable Before & After Transformation

Dealing With Missing Values (NAs)

First, we dealt with features almost exclusively filled with NAs that had explicit meanings and did not represent missing observations. For instance, two features referred to pool quality and condition. However, only 7 houses had a pool while NAs in that column meant that there was no pool. In the same way, the NAs in the alley column meant that there was no alley access, NAs in the fence column meant that there was no fence, and the NAs in fireplace column meant there was no fireplace.

Exhibit 3. Missing Values in Features

When imputation wasn’t feasible due to lack of information, we dropped the feature. When imputation was feasible, we either replaced the NA with its actual meaning, or we imputed using mean, mode, or 0.

Outlier Removal 

Another way to reduce the skewness is by removing outliers. This procedure is not an easy task and having domain knowledge is usually an advantage. One way to remove outliers is by deleting observations of features which are three or more standard deviations from the mean. However, in our case, this would remove hundreds of observations from the dataset. To avoid such a huge data loss, we created a scatter plot of sale price versus other features and decided to remove the four most apparent outliers.

Exhibit 4. Outliers in the Scatter Plot

Feature Engineering & Selection

Feature engineering is fundamental to the application of machine learning, so we individually analyzed features and discussed their relevance to this project.

For categorical data, we observed each features' relationship by comparing the mean and median sales prices of the different groups using ANOVA and boxplots respectively.

Exhibit 5. Boxplot of Overall Quality vs. Sale Price

For numerical data, in order to avoid high multicollinearity, we examined regression plots for each numerical feature against sale price and created correlation heat map for all numerical features.

Exhibit 6. Sample Scatter Plot and Heat Map

Based on our results from the previous steps, we decided to drop features irrelevant to the sales price as well as features with high correlation. For example, ground living area square feet was equal to the sum of first-floor square feet, second-floor square feet, and low quality finished square feet. Thus, we decided to drop the last three and only keep ground living area. We dropped a majority of the quality and condition columns because we assumed that “OverallCondition” and “OverallQuality” described all these other features.

We also replaced ordinal categorical features with corresponding integers. For instance, we encoded the lot shape feature as follows: A regular shape was assigned the value of 4, slightly irregular a 3, moderately irregular a 2, and irregular a 1.

Lastly, we added a new feature:  "YrSinceRemod". This new feature equaled to year sold minus remodel date, thereby capturing the number of years since a house was either last remodeled or built. Moreover, instead of putting in all the bathroom features separately, we created a combination value named “Total bathrooms” that included basement full bathrooms, full bathrooms, basement half bathrooms, and half bathrooms. 


To find the best fit for our data, we decided to apply both linear and non-linear algorithms:

Exhibit 7. Modeling Workflow

Linear Regression (Ridge, Lasso, ElasticNet)

We decided to begin our modeling process linear models after normalizing the independent and dependent variables.

The first model out of the linear regressions we used was the multiple linear regression model. The root mean squared error (RMSE) for the multiple linear regression was 0.18. The reason the error was so high was most likely because this model has no penalization for the number of features used and therefore is prone to multicollinearity issues. 

After that, we tried the penalized regression models: Lasso, Ridge, and ElasticNet. For each of the models, several neighborhood features received a significant importance in the feature importance output, suggesting that the neighborhood greatly impacts the sale price.

Exhibit 8. Feature Importance from Lasso

Out of all the regressions we implemented, lasso regression achieved the best RMSE, 0.124, probably due to actually dropping features, which a ridge regression does not do. This is visualized in exhibit 9 that shows how a lasso regression shrinks the coefficients to zero while the features in a ridge regression only approach but never equal zero.

Exhibit 9. Coefficients Selection in Lasso (top) and Ridge (bottom)

As can be seen in exhibit 10 below, most predictions from our lasso regression were fairly accurate.

Exhibit 10. Lasso Regression Model Result

Our ridge regression, on the other hand, resulted in the worst result out of the regressions we implemented with an RMSE of 0.138.

ElasticNet, a combination of both ridge and lasso, achieved the second best RMSE with 0.132. This makes sense as the RMSE is just in between the RMSEs of its components: ridge and lasso regression.  

Non-Linear Algorithms (Random Forest, XGBoost)

Random Forest

Decision Tree models, such as random forest, were also investigated in this project. Random forest is one of the most popular machine learning algorithms nowadays. It is reliable, effective, and very forgiving. Unlike linear regression, random forest is not sensitive to outliers. Consequently, it is not necessary to normalize the data or dummify categorical features. The only drawback of using it is that random forest requires more computation than traditional machine learning algorithms.

Exhibit 11 shows the feature importance result of random forest. As can be seen from the figure, the “OverallQual” is the dominant feature in random forest model, followed by “‘GrLivArea” and “GarageCar.”

Exhibit 11. Feature Importance from Random Forest

Note that the result is quite different from that of linear regression. This difference stems from the different interpretations of feature importance. The importance of the features in random forest is determined by the features’ contribution to the drop in the aggregated RSS (for regressions) while in linear regression, feature importance is generally indicated by the magnitude of coefficients.

Although random forest performs better than linear regression models in most cases, it is not the best model for the data in this project. The RMSE of random forest model was 0.137, which is not as good as linear models’ result. This might be due to the following: first, the dataset is very linear, which fits the linear regression very well and second, the removal of outliers significantly improved linear models’ results, while random forest didn’t get any benefit from doing so.


Finally, we also decided to try XGBoost on our data. XGBoost is a gradient boosting algorithm that has gained lots of popularity in many data science competitions, especially on Kaggle. Benefits of XGBoost include optimization for speed and performance due to enabling parallel tree boosting. It outperforms most traditional algorithms in both, accuracy and computation time. For a more detailed evaluation of XGBoost’s performance measures, we recommend taking a look at this article.

First, let’s take a look at the XGBoost feature importances. Feature importance in XGBoost can be interpreted as the relative contribution of each feature for the model. In our case, there were some significant differences between XGBoost and Random Forest regarding feature importances. On one hand, “LotArea”, which wasn’t of major importance in our Random Forest model, took the number one spot as the most important feature. On the other hand, we also observed some similarities. For instance, “GrLiveArea” was the second most important feature in our RF and XGBoost model.

Exhibit 12. Feature Importance from XGBoost


To conclude, let’s take a look at a summary of our RMSE scores in comparison with our Kaggle submission scores:

Exhibit 13. Model Score Comparison

While XGBoost turned out to have the lowest RMSE (Root Mean Squared Error) with 0.125280 on our validation set, it did not achieve the best Kaggle score among all of our models. Interestingly, the less complex linear models, lasso and ridge regression, received significantly better results on Kaggle than they did on our validation set. In the end, our lasso regression turned out to be our best-performing model with a Kaggle Score of 0.12406.

Our main takeaway from this project is that a more complex algorithm does not necessarily yield better results than a  simpler algorithm. Ultimately, whether a model performs well or not depends on the data at hand. 

About Authors

Lukas Frei

Lukas Frei is an aspiring data scientist currently completing the 12-week bootcamp at the New York City Data Science Academy. Besides his passion for data, he is also very interested in business and finance. Lukas holds a BS...
View all posts by Lukas Frei >

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