Skip to main content

Advertisement

Log in

Performance evaluation of different machine learning techniques for prediction of heart disease

  • Original Article
  • Published:
Neural Computing and Applications Aims and scope Submit manuscript

Abstract

Heart diseases are of notable public health disquiet worldwide. Heart patients are growing speedily owing to deficient health awareness and bad consumption lifestyles. Therefore, it is essential to have a framework that can effectually recognize the prevalence of heart disease in thousands of samples instantaneously. At this juncture, the potential of six machine learning techniques was evaluated for prediction of heart disease. The recital of these methods was assessed on eight diverse classification performance indices. In addition, these methods were assessed on receiver operative characteristic curve. The highest classification accuracy of 85 % was reported using logistic regression with sensitivity and specificity of 89 and 81 %, respectively.

This is a preview of subscription content, log in via an institution to check access.

Access this article

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Fig. 1
Fig. 2
Fig. 3
Fig. 4
Fig. 5
Fig. 6
Fig. 7

Similar content being viewed by others

References

  1. Farran B, Channanath AM, Behbehani K, Thanaraj TA (2013) Predictive models to assess risk of type 2 diabetes, hypertension and comorbidity: machine-learning algorithms and validation using national health data from Kuwait—a cohort study. BMJ open 3(5):e002457

    Article  Google Scholar 

  2. Heydari M, Teimouri M, Heshmati Z, Alavinia SM (2015) Comparison of various classification algorithms in the diagnosis of type 2 diabetes in Iran. Int J Diabetes Dev Ctries 36(2):1–7

    Google Scholar 

  3. Bansal A, Agarwal R, Sharma R (2015) Determining diabetes using iris recognition system. Int J Diabetes Dev Ctries 35(4):432–438

    Article  Google Scholar 

  4. Kalaiselvi C, Nasira GM (2015) Classification and prediction of heart disease from diabetes patients using hybrid particle swarm optimization and library support vector machine algorithm. Int J Comput Algorithm 4(1):2278–2397

    Google Scholar 

  5. Bhramaramba R, Allam AR, Kumar VV, Sridhar G (2011) Application of data mining techniques on diabetes related proteins. Int J Diabetes Dev Ctries 31(1):22–25

    Article  Google Scholar 

  6. King RD (1992) Statlog databases. Department of Statistics and Modelling Science, University of Strathclyde, Glasgow

    Google Scholar 

  7. Bache K, Lichman M (2013) UCI machine learning repository [http://archive.ics.uci.edu/ml]. University of California, School of Information and Computer Science. Irvine, CA

  8. García-Pedrajas N, Hervás-Martínez C, Ortiz-Boyer D (2005) Cooperative coevolution of artificial neural network ensembles for pattern classification. Evolut Comput IEEE Trans 9(3):271–302

    Article  Google Scholar 

  9. Yao X, Liu Y (1998) Making use of population information in evolutionary artificial neural networks. Syst Man Cybern Part B: Cybern IEEE Trans 28(3):417–425

    Google Scholar 

  10. Bishop CM (1995) Neural networks for pattern recognition. Oxford University Press, Cambridge

    MATH  Google Scholar 

  11. Haykin S (2010) Neural networks: a comprehensive foundation, 1994. Mc Millan, New Jersey

    MATH  Google Scholar 

  12. Bahrammirzaee A (2010) A comparative survey of artificial intelligence applications in finance: artificial neural networks, expert system and hybrid intelligent systems. Neural Comput Appl 19(8):1165–1195

    Article  Google Scholar 

  13. Hoptroff RG (1993) The principles and practice of time series forecasting and business modelling using neural nets. Neural Comput Appl 1(1):59–66

    Article  Google Scholar 

  14. Azar AT (2013) Fast neural network learning algorithms for medical applications. Neural Comput Appl 23(3–4):1019–1034

    Article  Google Scholar 

  15. Brown MP, Grundy WN, Lin D, Cristianini N, Sugnet CW, Furey TS, Ares M, Haussler D (2000) Knowledge-based analysis of microarray gene expression data by using support vector machines. Proc Natl Acad Sci 97(1):262–267

    Article  Google Scholar 

  16. Furey TS, Cristianini N, Duffy N, Bednarski DW, Schummer M, Haussler D (2000) Support vector machine classification and validation of cancer tissue samples using microarray expression data. Bioinformatics 16(10):906–914

    Article  Google Scholar 

  17. Ranawana R, Palade V (2005) A neural network based multi-classifier system for gene identification in DNA sequences. Neural Comput Appl 14(2):122–131

    Article  Google Scholar 

  18. Yasdi R (2000) A literature survey on applications of neural networks for human-computer interaction. Neural Comput Appl 9(4):245–258

    Article  MATH  Google Scholar 

  19. Vapnik VN, Vapnik V (1998) Statistical learning theory, vol 2. Wiley, New York

    MATH  Google Scholar 

  20. Vapnik V (2000) The nature of statistical learning theory. Springer, Berlin

    Book  MATH  Google Scholar 

  21. Burges CJ (1998) A tutorial on support vector machines for pattern recognition. Data Min Knowl Disc 2(2):121–167

    Article  Google Scholar 

  22. Shafer G, Pearl J (1990) Readings in uncertain reasoning. Morgan Kaufmann Publishers Inc., Burlington

    MATH  Google Scholar 

  23. Heckerman D, Geiger D, Chickering DM (1995) Learning Bayesian networks: the combination of knowledge and statistical data. Mach Learn 20(3):197–243

    MATH  Google Scholar 

  24. Jensen FV (1996) An introduction to Bayesian networks, vol 210. UCL press, London

    Google Scholar 

  25. Peral J (1988) Probabilistic reasoning in intelligent systems. Morgan Kaufmann, San Mateo, Cali fornia 12:241–288

  26. Castillo E (1997) Expert systems and probabilistic network models. Springer, Berlin

    Book  Google Scholar 

  27. Hosmer DW Jr, Lemeshow S (2004) Applied logistic regression. Wiley, New York

    MATH  Google Scholar 

  28. Schumacher M, Roßner R, Vach W (1996) Neural networks and logistic regression: part I. Comput Stat Data Anal 21(6):661–682

    Article  MATH  Google Scholar 

  29. Vach W, Roßner R, Schumacher M (1996) Neural networks and logistic regression: part II. Comput Stat Data Anal 21(6):683–701

    Article  MATH  Google Scholar 

  30. Hajmeer M, Basheer I (2003) Comparison of logistic regression and neural network-based classifiers for bacterial growth. Food Microbiol 20(1):43–55

    Article  Google Scholar 

  31. Aha DW (1997) Lazy learning. Kluwer academic publishers, Berlin

    Book  MATH  Google Scholar 

  32. Kanmani S, Uthariaraj VR, Sankaranarayanan V, Thambidurai P (2007) Object-oriented software fault prediction using neural networks. Inf Softw Technol 49(5):483–492

    Article  Google Scholar 

  33. Geisser S (1993) Predictive inference, vol 55. CRC Press, Boca Raton

    Book  MATH  Google Scholar 

  34. Metz CE (1978) Basic principles of ROC analysis. In: Freeman LM (ed) Seminars in nuclear medicine. vol 4. Elsevier, pp 283–298

Download references

Acknowledgments

I am greatly thankful to Department of Biotechnology, New Delhi, for providing Bioinformatics Infrastructure Facility of DBT at Maulana Azad National Institute of Technology, Bhopal.

Author information

Authors and Affiliations

Authors

Corresponding author

Correspondence to Ashok Kumar Dwivedi.

Ethics declarations

Conflict of interest

None.

Rights and permissions

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

Cite this article

Dwivedi, A.K. Performance evaluation of different machine learning techniques for prediction of heart disease. Neural Comput & Applic 29, 685–693 (2018). https://doi.org/10.1007/s00521-016-2604-1

Download citation

  • Received:

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1007/s00521-016-2604-1

Keywords

Navigation