https://github.com/qtle3/simple-linear-regression
A Python script that implements a simple linear regression model to predict employee salaries based on years of experience, with predictions for mid-level and senior-level roles.
https://github.com/qtle3/simple-linear-regression
coefficients-and-intercept data-handling model-training prediction visualization
Last synced: 2 months ago
JSON representation
A Python script that implements a simple linear regression model to predict employee salaries based on years of experience, with predictions for mid-level and senior-level roles.
- Host: GitHub
- URL: https://github.com/qtle3/simple-linear-regression
- Owner: qtle3
- Created: 2024-08-11T00:03:32.000Z (about 1 year ago)
- Default Branch: main
- Last Pushed: 2024-08-12T04:47:40.000Z (about 1 year ago)
- Last Synced: 2025-04-08T20:34:24.092Z (6 months ago)
- Topics: coefficients-and-intercept, data-handling, model-training, prediction, visualization
- Language: Python
- Homepage:
- Size: 6.84 KB
- Stars: 1
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# Salary Prediction using Simple Linear Regression
This project implements a simple linear regression model to predict the salary of an employee based on their years of experience. The model is trained on a dataset containing years of experience and corresponding salaries, and is capable of predicting salaries for new data points.
## Detailed Summary
The script loads the salary dataset, splits it into training and test sets, and trains a linear regression model using `scikit-learn`. It visualizes the regression line against both training and test data to illustrate how well the model fits. The model predicts salaries for employees with 5 and 12 years of experience, rounding the results to two decimal places for clarity. Additionally, the script outputs the model's coefficients and intercept, which provide insights into the linear relationship between years of experience and salary.
## Key Features
- **Data Handling:** Utilizes `pandas` for loading and manipulating the dataset.
- **Model Training:** Uses `scikit-learn` to train a simple linear regression model on the data.
- **Prediction:** Predicts employee salaries based on years of experience.
- **Visualization:** Visualizes the regression line against the actual data for both training and test sets.
- **Coefficients & Intercept:** Outputs the model's coefficients and intercept to understand the linear relationship.