File size: 1,528 Bytes
08c2460 64fc6eb 08c2460 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 | ---
license: apache-2.0
language:
- en
---
# HR Attrition Prediction API - Futurisys
This project provides a professional-grade REST API designed to predict employee attrition for **Futurisys**.
It uses a Machine Learning pipeline to analyze employee data and provide actionable insights for HR departments.
## Project Overview
The objective is to identify employees at risk of leaving the company by analyzing HR features.
**Key Features:**
- **Machine Learning Pipeline:** A robust model (Gradient Boosting/Random Forest) integrated with automated preprocessing.
- **FastAPI Framework:** High-performance API with built-in validation and asynchronous support.
---
## Project Structure
```text
.
βββ app/
β βββ main.py # Core API logic and Pydantic schemas
β βββ pipeline_rh.joblib # Serialized Scikit-Learn pipeline (Model + Scalers)
βββ notebooks/ # Research, EDA, and model training notebooks
βββ .gitignore # Ensures clean version control by ignoring temp files
βββ requirements.txt # List of Python dependencies
βββ README.md # Project documentation
```
## Installation & Setup
1. Prerequisites
Python 3.8+
Git
3. Clone the Repository
git clone <your-repository-url>
cd <your-project-folder>
3. Install dependencies
pip install -r requirements.txt
## Usage
### Running the API
## API Endpoints
GET /
POST /predict
## Author
Kevin L. - Data Science & Machine Learning Student |