Comparación de Algoritmos Machine Learning para la Predicción de Pacientes con Sospecha de COVID-19

Daniel Andrade-Girón; Edgardo Carreño-Cisneros; Cecilia Mejía-Dominguez; William Marín-Rodriguez; Henry Villarreal-Torres

doi:10.56294/saludcyt2023336

Original

Published: 2023-03-23

DOI: https://doi.org/10.56294/saludcyt2023336

Comparison of Machine Learning Algorithms for Predicting Patients with Suspected COVID-19

Abstract

The coronavirus disease (COVID-19) outbreak has infected millions of people, causing a high death rate worldwide. Patients suspected of having COVID-19 are transferred to different health facilities, which has caused a saturation in care, for which it is necessary to have a prediction model to classify patients at high risk of clinical deterioration. The objective of the research was to compare classification algorithms based on automatic learning machines, for the prediction of clinical diagnosis in patients with COVID-19. 1000 records of patients with suspected SARS-CoV-2 infection who were admitted by the emergency service in health establishments in Peru were collected. After pre-processing the data and engineering the attributes, a sample of 700 records was determined. Models were designed and algorithms were compared: Logistic Regression, Support Vector Machine, Nearest Neighbors, Decision Tree, Random Forest, and Navie Bayes. The evaluation of the results of each algorithm was carried out using Accuracy, precision, sensitivity and Chohen's Kappa to know the degree of agreement between the prediction by the learning machine and the results of reality, that is, to what extent both results agree in their measurement. The algorithm that presented the best results was the Support Vector Machine and Random Forest, which predicted the patients with an accuracy of 97%, and Cohen's Kappa of 0.95, with figures higher than the other models evaluated.

Keywords:

machine learning,

classification,

COVID-19,

epidemic,

morbidity,

How to Cite

Andrade-Girón D, Carreño-Cisneros E, Mejía-Dominguez C, Marín-Rodriguez W, Villarreal-Torres H. Comparison of Machine Learning Algorithms for Predicting Patients with Suspected COVID-19. Salud, Ciencia y Tecnología [Internet]. 2023 Mar. 23 [cited 2024 Apr. 19];3:336. Available from: https://revista.saludcyt.ar/ojs/index.php/sct/article/view/336

Copyright Notice

The article is distributed under the Creative Commons Attribution 4.0 License. Unless otherwise stated, associated published material is distributed under the same licence.

Article metrics

Google scholar: See link

Metrics

Metrics Loading ...

Vol. 3 (2023)

See full issue

Revistas/Journals

Issue

About

Author Guidelines