This project aims to predict employee attrition using machine learning techniques. By analyzing various employee-related factors, I build a predictive model to help organizations identify potential attrition risks and take proactive measures
- For a full overview, you can download the report here.
The dataset is sourced from Kaggle and includes various features related to employee demographics, work experience, and job satisfaction.
- Perform exploratory data analysis (EDA) to understand key attrition drivers.
- Train and evaluate machine learning models for attrition prediction.
- Provide insights into factors contributing to employee attrition.
Feature | Description |
---|---|
Employee ID | A unique identifier assigned to each employee. |
Age | The age of the employee, ranging from 18 to 60 years. |
Gender | The gender of the employee. |
Years at Company | The number of years the employee has been working at the company. |
Monthly Income | The monthly salary of the employee, in dollars. |
Job Role | The department or role the employee works in, encoded into categories such as Finance, Healthcare, Technology, Education, and Media. |
Work-Life Balance | The employee's perceived balance between work and personal life: (Poor, Below Average, Good, Excellent). |
Job Satisfaction | The employee's satisfaction with their job: (Very Low, Low, Medium, High). |
Performance Rating | The employee's performance rating: (Low, Below Average, Average, High). |
Number of Promotions | The total number of promotions the employee has received. |
Overtime | Whether the employee works overtime: (Yes or No). |
Distance from Home | The distance between the employee's home and workplace, in miles. |
Education Level | The highest education level attained by the employee: (High School, Associate Degree, Bachelor’s Degree, Master’s Degree, PhD). |
Marital Status | The marital status of the employee: (Divorced, Married, Single). |
Number of Dependents | The number of individuals financially dependent on the employee. |
Job Level | The job level of the employee: (Entry, Mid, Senior). |
Company Size | The size of the company the employee works for: (Small, Medium, Large). |
Company Tenure | The total number of years the employee has been working in the industry. |
Remote Work | Whether the employee works remotely: (Yes or No). |
Leadership Opportunities | Whether the employee has leadership opportunities: (Yes or No). |
Innovation Opportunities | Whether the employee has opportunities for innovation: (Yes or No). |
Company Reputation | The employee's perception of the company's reputation: (Very Poor, Poor, Good, Excellent). |
Employee Recognition | The level of recognition the employee receives: (Very Low, Low, Medium, High). |
Target Feature | Description |
---|---|
Attrition | Whether the employee has left the company, encoded as 0 (Stayed) and 1 (Left). |