Cardiovascular Disease Detection Using Machine Learning

Autores/as

  • Rodrigo Ibarra Universidad Panamericana
  • Jaime León Universidad Panamericana
  • Iván Ávila Universidad Panamericana
  • Hiram Ponce Universidad Panamericana

DOI:

https://doi.org/10.13053/cys-26-4-4422

Palabras clave:

Machine learning, classification, heart disease

Resumen

The detection of Cardiovascular Diseases (CVDs) prematurely is of great interest for the Healthcare Industry. According to the World Health Organization, heart diseases represent 32% of global deaths by 2019. In this work, we propose building an interpretable machine learning model to detect CVDs. For this, we use a public dataset consisting of over 320 thousand records and 279 features. We explore the performance of three well-known classifiers and we build them using hyper-parameter techniques. For interpretability, feature relevance is tested. After the experimental results, we found Random Forest to performed the best with 94% of accuracy and 81% of area under the ROC curve. We also implement an easy web application as a tool for detecting CVDs using relevant features information.

Biografía del autor/a

Rodrigo Ibarra, Universidad Panamericana

Facultad de Ingeniería

Jaime León, Universidad Panamericana

Facultad de Ingeniería

Iván Ávila, Universidad Panamericana

Facultad de Ingeniería

Hiram Ponce, Universidad Panamericana

Facultad de Ingeniería

Descargas

Publicado

2022-12-25

Número

Sección

Artículos