https://github.com/rahat-khan-pathan/salary-prediction-using-simple-linear-regression

This model can predict salary from years of experience
https://github.com/rahat-khan-pathan/salary-prediction-using-simple-linear-regression

linear-regression machine-learning python

Last synced: about 2 months ago
JSON representation

This model can predict salary from years of experience

• Host: GitHub
• URL: https://github.com/rahat-khan-pathan/salary-prediction-using-simple-linear-regression
• Owner: Rahat-Khan-Pathan
• License: mit
• Created: 2022-11-24T23:10:21.000Z (about 2 years ago)
• Default Branch: main
• Last Pushed: 2022-11-26T06:53:22.000Z (about 2 years ago)
• Last Synced: 2023-03-10T18:12:10.235Z (almost 2 years ago)
• Topics: linear-regression, machine-learning, python
• Language: Python
• Homepage:
• Size: 38.1 KB
• Stars: 5
• Watchers: 1
• Forks: 1
• Open Issues: 0
• Metadata Files:
  • Readme: readme.md
  • License: LICENSE.md

Awesome Lists containing this project

README

        
# 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

```

![Figure of regression](https://github.com/Rahat-Khan-Pathan/salary-prediction-using-simple-linear-regression/blob/main/figures/regression.png?raw=true)

R-Squared gained - 0.998 (Quite Good!)

See the Demo Video

Dataset Here