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

被引:5
作者
Velswamy, Karunakaran [1 ]
Velswamy, Rajasekar [2 ]
Swamidason, Iwin Thanakumar Joseph [1 ]
Chinnaiyan, Selvan [3 ]
机构
[1] Karunya Inst Technol & Sci, Coimbatore, Tamil Nadu, India
[2] SRM Inst Sci & Technol, Vadapalani Campus, Chennai, Tamil Nadu, India
[3] Natl Inst Technol, Trichy, Tamil Nadu, India
关键词
Classification; Feature selection; Bee algorithm; Support vector machine; Naive Bayes and KNN;
D O I
10.1007/s00500-021-06330-y
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
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, Naive 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 Naive Bayes and KNN.
引用
收藏
页码:13049 / 13057
页数:9
相关论文
共 20 条
[1]  
Alotaibi Fahd Saleh., 2019, Int. J. Adv. Comput. Sci. Appl, V10, P6, DOI [DOI 10.14569/IJACSA.2019.0100637, 10.14569/ijacsa.2019.0100637]
[2]  
Bhoyar S, 2021, 2021 2 INT C EMERGIN, P1, DOI DOI 10.1109/INCET51464.2021.9456389
[3]  
Garate-Escamila Anna Karen, 2020, Informatics in Medicine Unlocked, V19, P193, DOI 10.1016/j.imu.2020.100330
[4]   Prediction of heart disease by classifying with feature selection and machine learning methods [J].
Gazeloglu, Cengiz .
PROGRESS IN NUTRITION, 2020, 22 (02) :660-670
[5]  
Gill, 2021, P 2021 5 INT C COMP, P1779, DOI DOI 10.1109/ICCMC51019.2021.9418345
[6]   COMPARATIVE ANALYSIS OF SUPERVISED MACHINE LEARNING ALGORITHMS FOR HEART DISEASE DETECTION [J].
Huapaya, Hector Daniel ;
Rodriguez, Ciro ;
Esenarro, Doris .
3C TECNOLOGIA, 2020, :233-247
[7]   Early Prediction of Heart Disease Using PCA and Hybrid Genetic Algorithm with k-Means [J].
Islam, Md Touhidul ;
Rafa, Sanjida Reza ;
Kibria, Md Golam .
2020 23RD INTERNATIONAL CONFERENCE ON COMPUTER AND INFORMATION TECHNOLOGY (ICCIT 2020), 2020,
[8]  
Karunakaran V., 2020, International Journal of Enterprise Network Management, V11, P32
[9]  
Karunakaran V., 2019, International Journal of Civil Engineering and Technology (IJCIET), V10, P1745
[10]   Heart Disease Prediction using Hybrid machine Learning Model [J].
Kavitha, M. ;
Gnaneswar, G. ;
Dinesh, R. ;
Sai, Y. Rohith ;
Suraj, R. Sai .
PROCEEDINGS OF THE 6TH INTERNATIONAL CONFERENCE ON INVENTIVE COMPUTATION TECHNOLOGIES (ICICT 2021), 2021, :1329-1333