Skip to content

The goal is to predict whether an employee's salary exceeds a specific threshold (e.g., $50,000) based on various personal and professional attributes such as age, education level, occupation, hours worked per week, and years of experience.

Notifications You must be signed in to change notification settings

Naveen-Yerrannagari/Employee-Salary-Prediction-

Folders and files

NameName
Last commit message
Last commit date

Latest commit

ย 

History

7 Commits
ย 
ย 
ย 
ย 
ย 
ย 

Repository files navigation

๐Ÿง  Employee Salary Prediction using Machine Learning

Welcome to the Employee Salary Prediction project!
This machine learning model predicts whether an employee's salary is above or below $50K annually, based on their personal and professional attributes.


๐Ÿ“Œ Project Highlights

  • ๐Ÿ” Data Source: UCI Adult Income Dataset
  • โš™ Model Type: Supervised Classification
  • ๐Ÿ“ˆ Algorithms: Logistic Regression, Random Forest, Gradient Boosting, SVM, KNN
  • ๐ŸŒ Deployment Option: Streamlit / Flask
  • ๐Ÿงฉ Output: Predicts if salary is >50K or <=50K

๐Ÿ—‚ Folder Structure


โ”œโ”€โ”€ data/                  # Raw dataset
โ”œโ”€โ”€ notebooks/             # EDA & model building notebooks
โ”œโ”€โ”€ requirements.txt
โ””โ”€โ”€ README.md


๐Ÿงฐ Technologies Used

Tool/Library Purpose
Python Programming Language
Pandas Data Manipulation
NumPy Numerical Computation
Scikit-learn ML Algorithms & Preprocessing
Matplotlib Data Visualization
Joblib Model Serialization
Streamlit Web App Deployment (Optional)

๐Ÿ“Š Features Used

  • Age
  • Education Level
  • Occupation
  • Hours per Week
  • Work Experience
  • Native Country
  • Marital Status
  • Salary Label (>50K or <=50K)

๐Ÿš€ How to Run the Project

  1. Clone the repository
    bash git clone https://github.com/Naveen-Yerrannagari/Employee-Salary-Prediction-.git cd Employee-Salary-Prediction

  2. Install the dependencies

    bash pip install -r requirements.txt

  3. Train the model

    bash python src/train_model.py

  4. Launch the Streamlit app (Optional)

    bash streamlit run app.py


๐Ÿ“ˆ Model Performance

  • โœ… Accuracy: 85%+ (varies by algorithm)
  • ๐Ÿ“‰ Precision, Recall, F1-score: Reported in terminal output
  • ๐Ÿ“Š Confusion Matrix & Feature Importance: Visualized in notebook

๐Ÿ” Example Prediction

Feature Value
Age 34
Education Bachelors
Occupation Tech-support
Hours/Week 40
Experience 5
Prediction ๐Ÿ’ฐ <=50K

๐Ÿง  Future Improvements

  • ๐Ÿ“ฆ Add advanced models like XGBoost or LightGBM
  • ๐Ÿงฎ Include regression mode to predict actual salary amount
  • ๐ŸŒ Add country-wise salary normalization
  • ๐Ÿงพ Improve explainability using SHAP / LIME

๐Ÿ“š References


๐Ÿค Contributing

Contributions, bug reports, and feature requests are welcome! Please open an issue or submit a pull request.

---# Employee-Salary-Prediction- The goal is to predict whether an employee's salary exceeds a specific threshold (e.g., $50,000) based on various personal and professional attributes such as age, education level, occupation, hours worked per week, and years of experience.

About

The goal is to predict whether an employee's salary exceeds a specific threshold (e.g., $50,000) based on various personal and professional attributes such as age, education level, occupation, hours worked per week, and years of experience.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published