Employee Burnout Prediction

This project aims to predict employee burnout using various machine learning models. The dataset used in this project contains information about employees and their burnout rates.

Getting Started

Prerequisites

• Python 3.x
• Virtual environment (optional but recommended)

Installation

1. Clone the repository:

   git clone <repository-url>
   cd <repository-directory>

2. Create and activate a virtual environment:

   python -m venv venv
   source venv/bin/activate  # On Windows use `venv\Scripts\activate`

3. Install the required packages:

   pip install -r requirements.txt

Dataset

Download the dataset from this link and place the CSV files in the data/ directory with the name as EmployeeBurnout_Raw.csv.

Running the Project

1. Exploration:

   python src/01_exploration.py

2. Cleaning:

   python src/02_cleaning.py

3. Processing:

   python src/03_processing.py

4. Training:

   python src/04_training.py

License

This project is licensed under the MIT License.