The heart plays a vital role in living organisms. Heart-related disorders are more difficult to diagnose and predict, and errors in diagnosis and prognosis can have fatal consequences. A subfield of artificial intelligence, machine learning uses training from natural occurrences to predict any type of event with minimal support. In this work, we estimate the predictive power of machine learning algorithms that use several feature selection strategies to forecast heart disease. These algorithms include Select k best, Mutual Information Score, Recursive Feature Elimination (RFE), Support Vector Machine (SVM) estimator, and cross-validation. Random forest and gradient boosting were the predictive models used on the Framingham dataset for training and testing. This study uses the SMOTE technique for class imbalance problems. In our research, we got the highest accuracy for select k best feature selection modeled using random forest with an accuracy of 95.61% and gradient boosting with an accuracy of 95.43% considering hyperparameters.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Comparative Study of Feature Selection Algorithms for Heart Disease Prediction

  • Preetam Vallabhaneni,
  • Aditya Dudugu,
  • Harishitha Chowdary Alapati,
  • Sravya Meda,
  • Ashok Kumar Pradhan

摘要

The heart plays a vital role in living organisms. Heart-related disorders are more difficult to diagnose and predict, and errors in diagnosis and prognosis can have fatal consequences. A subfield of artificial intelligence, machine learning uses training from natural occurrences to predict any type of event with minimal support. In this work, we estimate the predictive power of machine learning algorithms that use several feature selection strategies to forecast heart disease. These algorithms include Select k best, Mutual Information Score, Recursive Feature Elimination (RFE), Support Vector Machine (SVM) estimator, and cross-validation. Random forest and gradient boosting were the predictive models used on the Framingham dataset for training and testing. This study uses the SMOTE technique for class imbalance problems. In our research, we got the highest accuracy for select k best feature selection modeled using random forest with an accuracy of 95.61% and gradient boosting with an accuracy of 95.43% considering hyperparameters.