CVD / README.md
yonghan93's picture
Update README.md
e9817b5 verified
|
raw
history blame
1.11 kB
metadata
title: CVD Risk Predictor
sdk: streamlit
emoji: πŸš€
colorFrom: indigo
colorTo: red
pinned: true

Cardiovascular Disease Risk Predictor

This is a Streamlit-based web app to predict cardiovascular disease (CVD) risk using a trained FT-Transformer model.

🧠 Model

The model is trained on health data with features like age, blood pressure, cholesterol, and lifestyle factors. It uses a decision threshold of 0.38 to determine high risk.

πŸš€ How to Use

  1. Clone this repository or upload to Hugging Face Spaces.
  2. Ensure the folder structure is preserved.
  3. The app will automatically launch using streamlit run app.py.

πŸ“ Folder Structure

FTTransformerApp/
β”œβ”€β”€ app.py
β”œβ”€β”€ requirements.txt
└── FTTransformerModel/
    β”œβ”€β”€ model.ckpt
    β”œβ”€β”€ config.yml
    β”œβ”€β”€ encoder.pkl
    β”œβ”€β”€ threshold.json
    β”œβ”€β”€ callbacks.sav
    β”œβ”€β”€ custom_params.sav
    β”œβ”€β”€ datamodule.sav
    └── exp_logger.sav

βœ… Prediction Threshold

This app uses a probability threshold of 0.38 to classify as high risk for CVD.