Disease Classifier (XGBoost)
A multi-class disease classification model built with XGBoost that predicts diseases based on patient symptoms, age, and gender.
Model Overview
| Property | Value |
|---|---|
| Algorithm |  |
| Objective |  |
| Task |  |
| Input |  |
| Output |  |
🚀 Quick Start
Installation
pip install huggingface_hub joblib pandas numpy xgboost scikit-learn
Load and Use the Model
import joblib
import pandas as pd
import numpy as np
from huggingface_hub import hf_hub_download
# Download model from Hugging Face
REPO_ID = "ARUNAGIRINATHAN/disease-classifier"
model = joblib.load(hf_hub_download(REPO_ID, "xgboost_model.joblib"))
label_encoder = joblib.load(hf_hub_download(REPO_ID, "label_encoder.joblib"))
feature_columns = joblib.load(hf_hub_download(REPO_ID, "feature_columns.joblib"))
Made with ❤️ for healthcare AI research