CVD & HT Risk Prediction Models
Machine learning models for cardiovascular disease and hypertension risk prediction.
Performance
| Model | Accuracy | F1-Score | Features |
|---|---|---|---|
| Random Forest CVD | 73-75% | 72-74% | 12 |
| Random Forest HT | 85-95% | 82-92% | 13 |
| XGBoost CVD | 73-75% | 72-74% | 12 |
| XGBoost HT | 85-95% | 82-92% | 13 |
Quick Start
import joblib
from huggingface_hub import hf_hub_download
# Download model
model_path = hf_hub_download(
repo_id="BrawnyCucumber/cvd-ht-prediction",
filename="random_forest_cvd.pkl"
)
# Load and use
model = joblib.load(model_path)
prediction = model.predict(X)
Models Included
random_forest_cvd.pkl- CVD predictionrandom_forest_ht.pkl- HT predictionxgboost_cvd.pkl- CVD predictionxgboost_ht.pkl- HT predictionfeature_selector_cvd.pkl- Feature selectionfeature_selector_ht.pkl- Feature selection
License
MIT
Inference Providers
NEW
This model isn't deployed by any Inference Provider.
🙋
Ask for provider support