/ft_linear_regression

This project is about AI and Machine Learning. A simple, basic machine learning algorithm. A program that predicts the price of a car by using a linear function train with a gradient descent algorithm.

Primary LanguagePython

ft_linear_regression - An introduction to machine learning

Summary: In this project, you will implement your first machine learning algorithm.


Usage

  1. Prediction before training
python3 predict.py
  • output
Type a Km to estimate price: 3421
Estimated price for km: 3421 is 0.
  1. Train
  • It generates model.csv which contains optimized theta values
python3 train.py
  1. Prediction after training
python3 predict.py
  • output
Type a Km to estimate price: 3421
Estimated price for km: 3421 is 8426.
  1. Evaluation
  • Evaluate the model using r2 score
python3 evaluate.py
  1. Visualization
python3 plot.py

Visualization

Before Training After Training
temp_1 temp_24

Animated Training Process

scatter_regression