Classification model for heart disease prediction with feature selection through modified bee algorithm

被引:0
作者
Karunakaran Velswamy
Rajasekar Velswamy
Iwin Thanakumar Joseph Swamidason
Selvan Chinnaiyan
机构
[1] Karunya Institute of Technology and Sciences,
[2] SRM Institute of Science and Technology,undefined
[3] Vadapalani Campus,undefined
[4] National Institute of Technology,undefined
来源
Soft Computing | 2022年 / 26卷
关键词
Classification; Feature selection; Bee algorithm; Support vector machine; Naïve Bayes and KNN;
D O I
暂无
中图分类号
学科分类号
摘要
Nowadays, a healthcare field produces a huge amount of data; for processing those data, some efficient techniques are required. In this paper, a classification model is developed for heart disease prediction and the attribute selection is carried out through a modified bee algorithm. The prediction of heart disease through models will help the practitioners to make a precise decision about patient health. Heart disease dataset is obtained from the UCI repository. Dataset consists of 76 features and all those seventy-six features have not contributed equal information during the time of classification. In the entire attributes, some of the attributes will contribute a large amount of information at the time of classification task and some of the attributes will contribute only a small amount of information during the classification task. In this paper, a modified bee algorithm is used to identify the best subset of attributes from the entire features in the dataset; i.e., the training phase of classification only retains those features that are contributing more information during classification task and it will reduce the training time of classifiers. The experiment is analyzed with an obtained reduced subset of attributes by using the following classifiers such as support vector machine, Naïve Bayes and KNN. The experimental result shows that the support vector machine classifier will provide a good classification accuracy, true positive rate, true negative rate, false positive rate and false negative rate compared to Naïve Bayes and KNN.
引用
收藏
页码:13049 / 13057
页数:8
相关论文
共 31 条
[1]  
Alotaibi FS(2019)Implementation of machine learning model to predict heart failure disease IJACSA Int J Adv Comput Sci Appl 10 6-670
[2]  
Garate-Escamilla AK(2020)Classification models for heart disease prediction using feature selection and PCA Inform Med Unlock 19 100330-64
[3]  
Hassani AHE(2020)Prediction of heart disease by classifying with feature selection and machine learning methods Prog Nutri 22 660-252
[4]  
Andres E(2019)A wrapper based feature selection approach using bees algorithm for extreme rainfall prediction via weather pattern recognition through SVM classifier Int J Civil Eng Technol IJCIET 10 1-662
[5]  
Gazeloğlu C(2020)Feature selection and instance selection using cuttlefish optimisation algorithm through tabu search Int J Enterp Netw Manag 11 32-366
[6]  
Karunakaran V(2019)Heart disease prediction and classification using machine learning algorithms optimized by particle swarm optimization and ant colony optimization Int J Intell Eng Syst 12 242-101
[7]  
Joseph SI(2019)Improving the accuracy of prediction of heart disease risk based on ensemble classification techniques Inform Med Unlock 16 100203-undefined
[8]  
Teja R(2020)Heart disease prediction using machine learning Int J Res Technol 9 659-undefined
[9]  
Suganthi M(2018)Prediction of heart disease using machine learning algorithms Int J Eng Technol 7 363-undefined
[10]  
Rajasekar V(2019)Instance selection and feature extraction using cuttlefish optimization algorithm and principal component analysis using decision tree Clust Comput 22 89-undefined