An end-to-end machine learning API that predicts insurance premiums based on user inputs.
The model is served using FastAPI and deployed as a Docker container for easy and consistent usage.
- Machine learning model for insurance premium prediction
- REST API built using FastAPI
- Input validation using Pydantic
- Dockerized for portability and easy deployment
- Auto-generated API documentation using Swagger UI
- The API returns the predicted premium category along with confidence scores and class-wise probabilities, accessible through FastAPI’s interactive Swagger UI.
- Python 3.11
- Scikit-learn
- Pandas, NumPy
- FastAPI
- Uvicorn
- Docker
docker pull bhumikab1/insurance-premium-api:latest