Skip to content

A linear regression model using gradient descent for home price prediction.

License

Notifications You must be signed in to change notification settings

Ismael-Deka/linear-reg-ML

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Linear Regression Model for House Price Prediction

This project implements a linear regression model to predict house prices based on various features.

Project Structure

The project contains the following files and directories:

  • data/: Directory containing the dataset file(s).
  • lin_utils.py: Utility module with functions for training and testing the linear regression model.
  • train.py: Script to train the linear regression model.
  • test.py: Script to test the trained model on new data.

Requirements

  • Python 3.x
  • pandas 1.2.4
  • numpy 1.23.5
  • matplotlib 3.5.1

Usage

  1. Model Training: Run the train.py script to train the linear regression model as well as prepare training data.

    python train.py
    
  2. Model Testing: After training the model, run the test.py script to test the trained model on new data and evaluate its performance.

    python test.py
    

Data

The project expects the dataset file data.csv to be present in the data/ directory. The dataset should include the following columns:

  • sqft_lot: Square footage of the lot.
  • sqft_living: Square footage of the living area.
  • bathrooms: Number of bathrooms.
  • bedrooms: Number of bedrooms.
  • condition: Condition rating of the house.
  • price: Sale price of the house.

The dataset was orginally sourced from Kaggle and is availible for download here.

Results

After training the model for 50 iterations with an initial learning rate of 2.0, and momentum coeffient of 0.9, the model achived a Mean Percent Difference of 28.9% when compared to true values in the training set and 27.1% for the test set(as of 05/29/2023)

Contributing

Contributions to this project are welcome. If you have any suggestions, improvements, or bug fixes, please submit a pull request.

License

This project is licensed under the MIT License.

About

A linear regression model using gradient descent for home price prediction.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages