Prediction of Heart Disease Based on Data Classification Models

With the increasing emphasis on healthy living, the prevention of major diseases has garnered widespread attention. Heart disease, as one of the leading causes of death globally, makes early prediction and identification extremely important. Currently, the prediction of heart disease mainly relies on a multidisciplinary approach. This article focuses on machine learning and selects four supervised learning classification models, including the Back Propagation (BP) neural network classification model, Genetic Algorithm (GA) -Back Propagation neural network classification model, Support Vector Machine (SVM) classification model and Random Forests Algorithm (RF) classification model, to evaluate their performance. Simulation experiments indicate that the GA-BP neural network classification model has the best generalization ability, while the RF classification model achieves the highest classification accuracy and recall rate, with an accuracy of 82.6% and a recall rate of 88.1% on the test set. Overall, the RF classification model performs the best among the four classification models. In future research, improvements and integrations of existing multiple data classification models will play a crucial role in enhancing classification accuracy.

Authors

Institutions

Publication Details

Journal
Applied and Computational Engineering
Published
2024-12-19
DOI
https://doi.org/10.54254/2755-2721/2025.18477
Primary Topic
Artificial Intelligence in Healthcare
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Prediction of Heart Disease Based on Data Classification Models

Yuanhang Yang
Applied and Computational Engineering
Artificial Intelligence in Healthcare
article

Prediction of Heart Disease Based on Data Classification Models

Yuanhang Yang
article en

Abstract

With the increasing emphasis on healthy living, the prevention of major diseases has garnered widespread attention. Heart disease, as one of the leading causes of death globally, makes early prediction and identification extremely important. Currently, the prediction of heart disease mainly relies on a multidisciplinary approach. This article focuses on machine learning and selects four supervised learning classification models, including the Back Propagation (BP) neural network classification model, Genetic Algorithm (GA) -Back Propagation neural network classification model, Support Vector Machine (SVM) classification model and Random Forests Algorithm (RF) classification model, to evaluate their performance. Simulation experiments indicate that the GA-BP neural network classification model has the best generalization ability, while the RF classification model achieves the highest classification accuracy and recall rate, with an accuracy of 82.6% and a recall rate of 88.1% on the test set. Overall, the RF classification model performs the best among the four classification models. In future research, improvements and integrations of existing multiple data classification models will play a crucial role in enhancing classification accuracy.

Applied and Computational EngineeringVol. 115(1)
Nanyang Technological University (SG)
Openalex Percentile: Top 67%
Artificial Intelligence in Healthcare
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.