Scientific Reports (Sep 2024)

Visualization obesity risk prediction system based on machine learning

  • Jinsong Du,
  • Sijia Yang,
  • Yijun Zeng,
  • Chunhong Ye,
  • Xiao Chang,
  • Shan Wu

DOI
https://doi.org/10.1038/s41598-024-73826-6
Journal volume & issue
Vol. 14, no. 1
pp. 1 – 9

Abstract

Read online

Abstract Obesity is closely associated with various chronic diseases.Therefore, accurate, reliable and cost-effective methods for preventing its occurrence and progression are required. In this study, we developed a visualized obesity risk prediction system based on machine learning techniques, aiming to achieve personalized comprehensive health management for obesity. The system utilized a dataset consisting of 1678 anonymized health examination records, including individual lifestyle factors, body composition, blood routine, and biochemical tests. Ten multi-classification machine learning models, including Random Forest and XGBoost, were constructed to identify non-obese individuals (BMI < 25), class 1 obese individuals (25 ≤ BMI < 30), and class 2 obese individuals (30 ≤ BMI). By evaluating the performance of each model on the test set, we selected XGBoost as the best model and built the visualized obesity risk prediction system based on it. The system exhibited good predictive performance and interpretability, directly providing users with their obesity risk levels and determining corresponding intervention priorities. In conclusion, the developed obesity risk prediction system possesses high accuracy and interactivity, aiding physicians in formulating personalized health management plans and achieving comprehensive and accurate obesity management.