Modelpublic

ARUNAGIRINATHAN/disease-classifier

sourceHugging Facemitupdated 9mo agoView on Hugging Face

3likes

Model Card

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

bash

pip install huggingface_hub joblib pandas numpy xgboost scikit-learn

Load and Use the Model

python

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