Unknown

Dataset Information

0

Predicting Low Cognitive Ability at Age 5-Feature Selection Using Machine Learning Methods and Birth Cohort Data.


ABSTRACT: Objectives: In this study, we applied the random forest (RF) algorithm to birth-cohort data to train a model to predict low cognitive ability at 5 years of age and to identify the important predictive features. Methods: Data was from 1,070 participants in the Irish population-based BASELINE cohort. A RF model was trained to predict an intelligence quotient (IQ) score ≤90 at age 5 years using maternal, infant, and sociodemographic features. Feature importance was examined and internal validation performed using 10-fold cross validation repeated 5 times. Results The five most important predictive features were the total years of maternal schooling, infant Apgar score at 1 min, socioeconomic index, maternal BMI, and alcohol consumption in the first trimester. On internal validation a parsimonious RF model based on 11 features showed excellent predictive ability, correctly classifying 95% of participants. This provides a foundation suitable for external validation in an unseen cohort. Conclusion: Machine learning approaches to large existing datasets can provide accurate feature selection to improve risk prediction. Further validation of this model is required in cohorts representative of the general population.

SUBMITTER: Bowe AK 

PROVIDER: S-EPMC9684182 | biostudies-literature | 2022

REPOSITORIES: biostudies-literature

altmetric image

Publications

Predicting Low Cognitive Ability at Age 5-Feature Selection Using Machine Learning Methods and Birth Cohort Data.

Bowe Andrea K AK   Lightbody Gordon G   Staines Anthony A   Kiely Mairead E ME   McCarthy Fergus P FP   Murray Deirdre M DM  

International journal of public health 20221110


<b>Objectives:</b> In this study, we applied the random forest (RF) algorithm to birth-cohort data to train a model to predict low cognitive ability at 5 years of age and to identify the important predictive features. <b>Methods:</b> Data was from 1,070 participants in the Irish population-based BASELINE cohort. A RF model was trained to predict an intelligence quotient (IQ) score ≤90 at age 5 years using maternal, infant, and sociodemographic features. Feature importance was examined and intern  ...[more]

Similar Datasets

| S-EPMC11334618 | biostudies-literature
| S-EPMC7804206 | biostudies-literature
| S-EPMC10217314 | biostudies-literature
| S-EPMC5481359 | biostudies-other
| S-EPMC8433864 | biostudies-literature
| S-EPMC6917601 | biostudies-literature
| S-EPMC10979063 | biostudies-literature
| S-EPMC7038475 | biostudies-literature
| S-EPMC8640070 | biostudies-literature
| S-EPMC8314163 | biostudies-literature