Predict Salary from Years of Experience using Simple Linear Regression

How to run this project

1. Install requirements from requirements.txt

$ pip install -r requirements.txt

2. Run train.py to train the model with 4900 data in salary.csv

$ python train.py

3. Run test.py to train the model with last 100 data in salary.csv

$ python test.py

4. Run main.py and choose the features to predict the probability.

$ python main.py

R-Squared gained - 0.998 (Quite Good!)

See the Demo Video

Dataset Here