Maternity Health Risk Prediction Models
This repository contains machine learning models for predicting maternity health risk levels.
Model Categories
Individual Models
- Decision Tree
- K-Nearest Neighbors (KNN)
- Logistic Regression
- Neural Network
- Random Forest
Ensemble Models
- Enhanced Ensemble
- Weighted Voting
Preprocessing Components
- Standard Scaler
- Initial Scaler
- Feature Selector
Usage
The models can be used through the Hugging Face Inference API. Each model can be accessed by specifying the model type in the API request.
Input Features
- Systolic Blood Pressure
- Diastolic Blood Pressure
- Blood Sugar
- BMI
- Heart Rate
- Previous Complications
- Preexisting Diabetes
- Gestational Diabetes
- Mental Health Status
Risk Levels
- Low Risk
- Medium Risk
- High Risk
