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 prediction
  • random_forest_ht.pkl - HT prediction
  • xgboost_cvd.pkl - CVD prediction
  • xgboost_ht.pkl - HT prediction
  • feature_selector_cvd.pkl - Feature selection
  • feature_selector_ht.pkl - Feature selection

License

MIT

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support