Predicting Survival of Heart Failure Patients Using Classification Algorithms

Main Article Content

Oladosu Oyebisi Oladimeji Olayanju Oladimeji


Heart failure is a situation that occurs when the heart is unable to pump enough blood to meet the needs of other organs in the body. It is responsible for the annual death of approximately 17 million people worldwide. Series of studies have been done to predict heart failure survival with promising results. Hence, the purpose of this study is to increase the accuracy of previous works on predicting heart failure survival by selecting significant predictive features in order of their ranking and dealing with class imbalance in the classification dataset. In this study, we propose an integrated method using machine learning. The proposed method shows promising results as it performs better than previous works and this study confirms that dealing with imbalanced dataset properly increases accuracy of a model. The model was evaluated based on metrics such as F-measure, Precision-Recall curve and Receiver Operating Characteristic Area Under Curve. This discovery has the potential to impact on clinical practice, when health workers aim at predicting if a patient will survive heart failure. Attention may be focused on mainly serum creatinine, ejection fraction, smoking status and age.


Download data is not yet available.

Article Details

How to Cite
Oladimeji, O., & Oladimeji, O. (2020, September 30). Predicting Survival of Heart Failure Patients Using Classification Algorithms. JITCE (Journal of Information Technology and Computer Engineering), 4(02), 90-94.


O.N. Emuoyibofarhe, S. Adebayo, A. Ibitoye, O.A. Madamidola and T. Aderibigbe, “Predictive System for Heart Disease Using a Machine Learning Trained Model,” International Journal of Computer (IJC), Vol. 34, No. 1, pp 140-152, 2019.

National Heart Lung and Blood Institute (NHLBI). Heart failure., [Online] Available: https:// Accessed 2020-04-10

D. Chicco and G. Jurman, “Machine learning can predict survival of patients with heart failure froms erum creatinine and ejection fractional one”, BMC Medical Informatics and Decision Making, Vol. 20, No. 16, 2019

T. Ahmad, A. Munir, S. H. Bhatti, M. Aflab, M.A Raza, “Survival analysis of heart failure patients: A case study”, vol. 12, no. 7, 2017, DOI: 10.1371/journal. pone.0181001

J. H. Joloudari et al, “Coronary Artery Disease Diagnosis; Ranking the Significant Features Using a Random Trees Model” International Journal of Environmental Research and Public Health, vol. 17, No 731,2020, DOI: 10.3390/ijerph17030731

WHO. Fact sheet on CVDs. Global Hearts. World Health Organization. 2016.

S. E. Awan, M. Bennamoun, F. Sohel, F.M. Sanfilippo, and G. Dwivedi, “Machine learning based prediction of heart failure readmission or death: implications of choosing the right model and the right metrics". ESC heart failure, Vol.6, No.2, pp 428-435, 2019.

C. M. Bishop., “Pattern Recognition and Machine Learning”. Springer ISBN 978-0-387-31073-2, 2006.

J. H. Joloudari, H. Saadatfar, A. Dehzangi, S. Shamshirband, “Computer-aided decision-making for predicting liver disease using PSO-based optimized SVM with feature selection”. Inform. Med. Unlocked 17, 100255, 2017.

G. Ahmad, M. A. Khan, S. Abbas, A. Athar, B. S. Khan, M. S. Aslam, “Automated Diagnosis of Hepatitis B Using Multilayer Mamdani Fuzzy Inference System”. J. Healthc. Eng. 2019, 6361318, 2019.

Y. Wang, L. Kung, S. Gupta, S. Ozdemir, “Leveraging big data analytics to improve quality of care in healthcare organizations: A configurational perspective”. Br. J. Manag. Vol.30, pp 362–388, 2019.

D. H. Smith, E. S. Johnson, M. L. Thorp, X. Yang, A. Petrik, R. W. Platt, and I. K. Crispel. “Predicting poor outcomes in heart failure.” Permanente J. Vol.15 No.4, pp4–11, 2011.

F. M.Zahid, S. Ramzan, S. Faisal, I. Hussain. “Gender based survival prediction models for heart failure patients: a case study in Pakistan.” PLoSONE. Vol. 14, No. 2, 2019.

J. Cai, J. Luo, S. Wang, and S. Yang. “Feature selection in machine learning: A new perspective.” Neurocomputing Vol. 300, pp. 70–79, 2018.

S. Mojrian, G. Pinter,J. Hassannataj Joloudari, I. Felde, N. Nabipour, L. Nadai, A. Mosavi, “Hybrid Machine Learning Model of Extreme Learning Machine Radial Basis Function for Breast Cancer Detection and Diagnosis: A Multilayer Fuzzy Expert System.” Preprints 2019, doi:10.20944/preprints201910.0349.v1
S. Liu, M. Motani, “Feature Selection Based on Unique Relevant Information for Health Data.” arXiv 2018, arXiv:1812.00415.

UCI, “Machine Learning Repository,” [Online] Available: Acccessed on 2020-04-02

S. O. Folorunsho and A. B. Adeyemo, “Alleviating classification problem of imbalanced dataset”, African Journal of Computing and ICT, Vol. 6, No. 1., pp. 137-144, 2013.

J. Crisóstomo, et al. “Hyperresistinemia and metabolic dysregulation: the close crosstalk in obese breast cancer,”. Endocrine, vol. 53, no. 2, pp. 433-442, 2016.

Weka, Available at: Accessed on 2020-04-03

B. W. Matthews. “Comparison of the predicted and observed secondary structure of T4 phage lysozyme.” Biochim Biophys Acta (BBA)-Protein Struct. Vol. 405 No. 2, pp. 442–51,1975.

T. Saito and M. Rehmsmeier. “The precision-recall plot is more informative than the ROC plot when evaluating binary classifiers on imbalanced datasets”. PLoSONE.Vol. 10, No. 3,0118432, 2015.

P. Diez, “Smart Wheelchairs and Brain-Computer Interfaces,” ScienceDirect, 2018