An end-to-end Machine Learning system that predicts whether a candidate should be Hired or Rejected based on their experience, skills, education, certifications, projects, salary expectation, and AI score.
The goal of this project is to assist recruiters in making data-driven hiring decisions using Machine Learning.
The system provides:
- β Hire / Reject prediction
- π Hiring probability
- π Model performance evaluation
- π Feature importance analysis
- π₯ Multiple candidate prediction
- π Prediction result export
Dataset
β
Data Cleaning
β
Exploratory Data Analysis
β
Feature Engineering
β
One-Hot Encoding
β
Train-Test Split
β
Logistic Regression
β
Model Evaluation
β
Model Saving
β
Candidate Prediction
- Python
- Pandas
- NumPy
- Matplotlib
- Seaborn
- Scikit-learn
- Joblib
- Logistic Regression
| Metric | Result |
|---|---|
| Training Samples | 800 |
| Testing Samples | 200 |
| Test Accuracy | 99.50% |
| Model | Logistic Regression |
- Experience
- Salary Expectation
- Projects Count
- AI Score
- Skills
- Education
- Certifications
- Job Role
The model analyzes which features have the strongest influence on hiring predictions.
The project includes:
feature_importance.csv- Top 10 Feature Importance visualization
git clone https://github.com/deepuchandel114-beep/Repository-name-AI-Based-Hiring-Prediction-System.git
cd Repository-name-AI-Based-Hiring-Prediction-Systempython -m venv venvWindows:
venv\Scripts\activatepip install -r requirements.txtpython hiring_prediction.pyThe system will train the model, evaluate its performance, save the trained model and encoder, perform candidate predictions, and generate result files.
AI-Based-Hiring-Prediction-System/
β
βββ Data/
β βββ hiring_data.csv
β
βββ screenshots/
β βββ AI Score vs Hiring Decision.png
β βββ Candidate Prediction Demo.png
β βββ Confusion Matrix.png
β βββ Education vs Hiring decision.png
β βββ Hiring decision distribution.png
β βββ Job vs Hiring Decision.png
β βββ Project Count vs Hiring decision.png
β βββ Salary Expectation vs Hiring Decision.png
β βββ Top 10 Feature Importance.png
β
βββ hiring_prediction.py
βββ hiring_model.pkl
βββ hiring_encoder.pkl
βββ feature_importance.csv
βββ prediction_result.csv
βββ multiple_candidate_results.csv
βββ requirements.txt
βββ README.md
βββ LICENSE
βββ .gitignore
The current system can be further enhanced with the following features:
- Build an interactive web interface using Streamlit or Flask
- Allow recruiters to enter candidate details through a form
- Add resume parsing and automatic candidate information extraction
- Integrate NLP-based skill matching
- Compare multiple Machine Learning algorithms
- Add model explainability using SHAP
- Store candidate predictions in a database
- Deploy the application on a cloud platform
- Add recruiter authentication and candidate management
- Develop a REST API for real-time hiring predictions
Deepak
B.Tech β Artificial Intelligence
This project was developed as an end-to-end Machine Learning project focused on applying predictive analytics to recruitment and hiring decisions.
This project is licensed under the MIT License.
This project is intended for educational and demonstration purposes.
The prediction generated by the model should not be used as the sole basis for real-world hiring decisions. Actual recruitment decisions should consider human evaluation, interviews, qualifications, experience, organizational requirements, and applicable fairness and employment policies.
The model's performance also depends on the quality, size, and characteristics of the training dataset.








