Prediction Model between Serum Vitamin D and Neurological Deficit in Cerebral Infarction Patients Based on Machine Learning

Hailiu Zhang,Guotao Yang,Aiqin Dong
DOI: https://doi.org/10.1155/2022/2914484
IF: 2.809
2022-06-29
Computational and Mathematical Methods in Medicine
Abstract:Objective. Vitamin D is associated with neurological deficits in patients with cerebral infarction. This study uses machine learning to evaluate the prediction model's efficacy of the correlation between vitamin D and neurological deficit in patients with cerebral infarction. Methods. A total of 200 patients with cerebral infarction admitted to the Department of Neurology of our hospital from July 2018 to June 2019 were selected. The patients were randomly divided into a training set ( n = 140 ) and a test set ( n = 60 ) in a 7 : 3 ratio. The prediction model is constructed from the training set's data, and the model's prediction effect was evaluated by test set data. The area under the receiver operator characteristic curve was used to assess the prediction efficiency of models. Results. In the training set, the area under the curve (AUC) of the logistic regression model and XGBoost algorithm model was 0.727 (95% CI: 0.601~0.854) and 0.818 (95% CI: 0.734~0.934), respectively. While in the test set, the AUC of the logistic regression model and XGBoost algorithm model was 0.761 (95% CI: 0.640~0.882) and 0.786 (95% CI: 0.670~0.902), respectively. Conclusion. The prediction model of the correlation between vitamin D and neurological deficit in patients with cerebral infarction based on machine learning has a good prediction efficiency.
mathematical & computational biology
What problem does this paper attempt to address?