Silva VC, Gorgulho B, Marchioni DM, Araujo TAD, Santos IDS, Lotufo PA, Benseñor IM. Clustering Analysis and Machine Learning algorithms in the prediction of dietary patterns: Cross-sectional results of the Brazilian Longitudinal Study of Adult Health (ELSA-Brasil).
J Hum Nutr Diet 2022;
35:883-894. [PMID:
35043491 DOI:
10.1111/jhn.12992]
[Citation(s) in RCA: 4] [Impact Index Per Article: 2.0] [Reference Citation Analysis] [Abstract] [Key Words] [Track Full Text] [Journal Information] [Subscribe] [Scholar Register] [Received: 07/25/2021] [Accepted: 01/11/2022] [Indexed: 11/30/2022]
Abstract
BACKGROUND
Machine learning investigates how computers can automatically learn. This study aimed to predict dietary patterns and compare algorithm performance in making predictions of dietary patterns.
METHODS
We analyzed the data of public employees (n=12,667) participating in the Brazilian Longitudinal Study of Adult Health (ELSA-Brasil). The K-means clustering algorithm and six other classifiers (support vector machines, naïve Bayes, K-nearest neighbors, decision tree, random forest, and xgboost) were used to predict the dietary patterns.
RESULTS
K-means clustering identified two dietary patterns. Cluster 1, labeled the Western pattern, was characterized by a higher energy intake and consumption of refined cereals, beans and other legumes, tubers, pasta, processed and red meats, high-fat milk and dairy products, and sugary beverages; Cluster 2, labeled the Prudent pattern, was characterized by higher intakes of fruit, vegetables, whole cereals, white meats, and milk and reduced-fat milk derivatives. The most important predictors were age, sex, per capita income, education level, and physical activity. The accuracy of the models varied from moderate to good (69-72%).
CONCLUSIONS
The algorithms' performance in dietary pattern prediction was similar, and the models presented may provide support in screener tasks and guide health professionals in the analysis of dietary data. This article is protected by copyright. All rights reserved.
Collapse