Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

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 22 hours ago
JSON representation

This model can predict salary from years of experience

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