metadata
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
.
├── 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
- Prerequisites
Python 3.8+ Git
- Clone the Repository
git clone cd
- Install dependencies
pip install -r requirements.txt
Usage
Running the API
API Endpoints
GET /
POST /predict
Author
Kevin L. - Data Science & Machine Learning Student