Skip to content

Repository files navigation

💼 Salary Prediction Classification

A Machine Learning project that predicts whether a person's salary belongs to a particular salary category based on demographic and employment-related features. The project includes data preprocessing, exploratory data analysis (EDA), feature engineering, model training, evaluation, and comparison of multiple classification algorithms.


📌 Project Overview

The objective of this project is to classify individuals into salary categories using machine learning techniques. The dataset is cleaned, analyzed, and transformed before training different classification models to identify the best-performing model.


🎯 Objectives

  • Perform data cleaning and preprocessing.
  • Conduct Exploratory Data Analysis (EDA).
  • Encode categorical variables.
  • Train multiple classification models.
  • Compare model performance using evaluation metrics.
  • Select the best model for salary prediction.

📂 Dataset

The dataset contains demographic and employment-related information such as:

  • Age
  • Workclass
  • Education
  • Marital Status
  • Occupation
  • Relationship
  • Race
  • Gender
  • Capital Gain
  • Capital Loss
  • Hours per Week
  • Native Country
  • Salary (Target Variable)

🛠️ Technologies Used

  • Python
  • Pandas
  • NumPy
  • Matplotlib
  • Seaborn
  • Scikit-learn
  • Google Colab / Jupyter Notebook

📊 Machine Learning Workflow

  1. Data Loading
  2. Data Cleaning
  3. Exploratory Data Analysis (EDA)
  4. Feature Encoding
  5. Train-Test Split
  6. Feature Scaling (if required)
  7. Model Training
  8. Model Evaluation
  9. Performance Comparison

🤖 Models Used

  • Logistic Regression
  • Decision Tree Classifier
  • Random Forest Classifier
  • K-Nearest Neighbors (KNN)
  • Support Vector Machine (SVM)
  • Naive Bayes (if implemented)

📈 Evaluation Metrics

The models are evaluated using:

  • Accuracy
  • Precision
  • Recall
  • F1-Score
  • Confusion Matrix
  • Classification Report

📁 Project Structure

Salary-Prediction-Classification/
│
├── Salary_Prediction_Classification.ipynb
├── README.md
└── dataset.csv

🚀 How to Run

  1. Clone the repository
git clone https://github.com/Ronak-si/Salary-Prediction-Classification.git
  1. Install the required libraries
pip install pandas numpy matplotlib seaborn scikit-learn
  1. Open the notebook
jupyter notebook Salary_Prediction_Classification.ipynb

or run it directly on Google Colab.


📌 Results

  • Cleaned and preprocessed the dataset.
  • Trained and compared multiple classification algorithms.
  • Evaluated model performance using standard classification metrics.
  • Identified the best-performing model for salary classification.

🔮 Future Improvements

  • Hyperparameter tuning using GridSearchCV.
  • Cross-validation for better model reliability.
  • Model deployment using Flask or Streamlit.
  • Feature importance visualization.
  • Interactive prediction dashboard.

👨‍💻 Author

Ronak Singh


⭐ If you found this project useful, don't forget to Star the repository!

Releases

Packages

Contributors

Languages