Hybrid approach to medical decision-making: prediction of heart disease with artificial neural network

Girish Shrikrushnarao Bhavekar, Pratiksha Vasantrao Chafle, Agam Das Goswami, Ganesh Kumar Marathula, Sumit Arun Hirve, Suraj Rajesh Karpe, Nitin Sonaji Magar, Amarsinh Baburao Farakte, Nileshchandra Kalbarao Pikle, Snehal Bankatrao Shinde, Amit Kamalakar Gaikwad

Abstract


Heart disease prediction is important in today’s world because it helps to reduce the unpredictable death rate of patients, and cardiac diseases are considered one of the most serious diseases affecting people. Hence, in this paper, a heart disease prediction model is designed for effective prediction of heart diseases by means of machine learning (ML) and deep learning (DL). This prediction uses the proposed method of an artificial neutral network and the Chi2 feature selection method applied to determine which features from the dataset were suitable for prediction. The proposed methodology uses classifiers like support vector machines (SVM), Naive Bayes (NB), logistic regression (LR), random forest (RF), and artificial neural networks (ANN). Python was used to conduct the study that assessed the ANN system proposal with the Cleveland heart disease dataset at the University of California (UCI). Compared to other algorithms, the model achieves an accuracy of 97.64% and takes 0.49 seconds to execute, making it superior in predicting heart disease.


Keywords


Cardiovascular; Feature selection; Machine learning; Neural network; Oversampling; Under sampling

Full Text:

PDF


DOI: https://doi.org/10.11591/eei.v13i6.5583

Refbacks

  • There are currently no refbacks.


Creative Commons License
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.

Bulletin of EEI Stats

Bulletin of Electrical Engineering and Informatics (BEEI)
ISSN: 2089-3191, e-ISSN: 2302-9285
This journal is published by the Institute of Advanced Engineering and Science (IAES) in collaboration with Intelektual Pustaka Media Utama (IPMU).