This project focuses on predicting data science salaries based on experience using Polynomial Regression. The dataset contains job-related details, including years of experience and corresponding salaries.
- Goal: Predict salaries in the data science field using machine learning.
- Algorithm: Polynomial Regression (Scikit-learn)
- Dataset: Synthetic or real-world salary data with experience as the main feature.
- Language: Python
- Python
- NumPy
- Pandas
- Matplotlib
- Scikit-learn
DS_Salary_Prediction_Polynomial.ipynbβ Jupyter Notebook with code and visualization.Position_Salaries.csvβ Dataset used for training the model.README.mdβ Project documentation.
- Data loading and exploration
- Polynomial feature transformation
- Model training and prediction
- Visualization of regression curve
- Model comparison (Linear vs Polynomial)