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

Downloads last month: -; Downloads are not tracked for this model. How to track

Inference Providers NEW

This model isn't deployed by any Inference Provider. 🙋 Ask for provider support