Prediction of Heart Disease Using Decision Tree in Comparison with Particle Swarm Optimization to Improve Accuracy

Ina Maryani, Rousyati, Indriyanti, Dany Pratmanto, Yustina Kristania, Mawadatul Maulidah

2023

Abstract

Heart disease is the leading cause of death worldwide. This disease can be prevented or treated easily if de-tected early. However, many people do not know the symptoms of heart disease, which results in delays in the treatment process. This disease can be caused by both modifiable and irreversible factors. This study aims to predict heart disease using the Naive Bayes and Decision Tree algorithms with and without the Particle Swarm Optimization (PSO) feature to predict heart disease. The results showed that the Decision Tree algorithm with the PSO feature provided the highest accuracy when predicting heart disease, with a value of 85.84%, 87.05% precision, 87.05% recall and an AUC value of 0.854. Whereas other algorithms such as Naive Bayes with PSO only provide an accuracy value of 85.73% and Decision Tree without PSO has an accuracy value of 83.23%, Naive Bayes without PSO has an accuracy value of 85.51%. Based on these results it can be concluded that a Decision Tree with PSO features is a more effective method for classifying heart disease compared to a Decision Tree without PSO, Naive Bayes without PSO and Naive Bayes with PSO. Therefore, it can be concluded that the Decision Tree algorithm with the PSO feature is the right choice for predicting heart disease.

Paper Citation

in Harvard Style

Maryani I., Rousyati., Indriyanti., Pratmanto D., Kristania Y. and Maulidah M. (2023). Prediction of Heart Disease Using Decision Tree in Comparison with Particle Swarm Optimization to Improve Accuracy. In Proceedings of the 3rd International Conference on Advanced Information Scientific Development - Volume 1: ICAISD; ISBN 978-989-758-678-1, SciTePress, pages 233-239. DOI: 10.5220/0012447300003848

in Bibtex Style

@conference{icaisd23,
author={Ina Maryani and Rousyati and Indriyanti and Dany Pratmanto and Yustina Kristania and Mawadatul Maulidah},
title={Prediction of Heart Disease Using Decision Tree in Comparison with Particle Swarm Optimization to Improve Accuracy},
booktitle={Proceedings of the 3rd International Conference on Advanced Information Scientific Development - Volume 1: ICAISD},
year={2023},
pages={233-239},
publisher={SciTePress},
organization={INSTICC},
doi={10.5220/0012447300003848},
isbn={978-989-758-678-1},
}

in EndNote Style

TY - CONF

JO - Proceedings of the 3rd International Conference on Advanced Information Scientific Development - Volume 1: ICAISD
TI - Prediction of Heart Disease Using Decision Tree in Comparison with Particle Swarm Optimization to Improve Accuracy
SN - 978-989-758-678-1
AU - Maryani I.
AU - Rousyati.
AU - Indriyanti.
AU - Pratmanto D.
AU - Kristania Y.
AU - Maulidah M.
PY - 2023
SP - 233
EP - 239
DO - 10.5220/0012447300003848
PB - SciTePress