five

Performance ML models.

收藏
Figshare2025-11-06 更新2026-04-28 收录
下载链接:
https://figshare.com/articles/dataset/Performance_ML_models_/30559131
下载链接
链接失效反馈
官方服务:
资源简介:
Background: Cardiovascular diseases (CVD) are one of the leading global causes of death, which requires an accurate early prediction. This study aimed to develop transparent machine learning (ML) models using National Health and Nutrition Examination Survey (NHANES) data from 2017–2023 to predict CVD risk based on dietary and health factors.Methods: We analyzed data from 12,382 adults (aged 18 and older) from NHANES 2017–2023, including 41 dietary, anthropometric, clinical, and demographic variables. Recursive Feature Elimination (RFE) was used to select an optimal subset of 30 predictors. To address substantial class imbalance in the outcome, we applied the Random Over-Sampling Examples (ROSE) technique to the training data. Five machine learning models—Logistic Regression, Random Forest, Support Vector Machines, XGBoost, and LightGBM—were trained and evaluated. Model interpretability was assessed using LIME and SHAP.Results: Participants with CVD differed significantly from those without CVD in age, waist circumference, systolic blood pressure, C-reactive protein (CRP), and multiple dietary nutrients, with a consistently lower nutrient intake in the CVD group. Among the ML models evaluated, XGBoost achieved the highest accuracy (0.8216) and recall (0.8645), while Random Forest showed the highest AUROC (0.8139). Interpretability analyses identified age as the strongest predictor, followed by vitamin B12, total cholesterol, CRP, and waist circumference.Conclusion: Interpretable ML models effectively identified key dietary and clinical factors for CVD risk. Nutrients like vitamin B12 and niacin, alongside established clinical indicators, emerged as significant predictors, underscoring their potential role in nutritional interventions and public health strategies for CVD prevention.
创建时间:
2025-11-06
5,000+
优质数据集
54 个
任务类型
进入经典数据集
二维码
社区交流群

面向社区/商业的数据集话题

二维码
科研交流群

面向高校/科研机构的开源数据集话题

数据驱动未来

携手共赢发展

商业合作