Machine learning model matters its accuracy: a comparative study of ensemble learning and AutoML using heart disease prediction

被引:0
|
作者
Rimal, Yagyanath [1 ]
Paudel, Siddhartha [2 ]
Sharma, Navneet [3 ]
Alsadoon, Abeer [4 ]
机构
[1] Pokhara Univ, Pokhara, Nepal
[2] IOE, Pulchowk Campus, Patan, Nepal
[3] IIS, Jaipur, India
[4] Charles Sturt Univ, Dubbo, Australia
关键词
Ensemble learning; Artificial neural network; Support vector machine; K nearest neighbors; Random forest; AutoML; HEALTH-CARE; SYSTEM;
D O I
10.1007/s11042-023-16380-z
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Ensemble machine learning is the concept of using multiple models to gain better performance from the combination of weak individual models. New researchers focus on improving machine learning models for accurate classification and prediction on test data, highlighting the critical issue of overall model quality. Once weak learners' ensembles for making strong models were compared separately, the precision, accuracy, and f1 model score were compared separately, and the majority of voting aggregation recommended the best mode for deployment. The model accuracy and their performances of the decision tree, logistic regression, support vector machine, random forest, artificial neural network, gaussian, k nearest neighbor, and multilayer perception were compared for the best model prediction. Similarly, Auto Machine Learning (AutoML) supports both binary classifications and regression problems that can be applied instantly without feature engineering directly. AutoML tries to develop a list of more robust models in tabular form and then determine whose accuracy prediction is the best. This research compares the eighteen (18) different machine learning models, i.e., eight (8) different models that were individually trained and ten (10) from AutoML, whose accuracy, mse, and r2 scores were compared with the same open-source heart disease data set. The support vector, logistic regression, and neural network models produced the highest 80% accuracy result compared to the gaussian, k nearest neighbors, and multilayer perception algorithms, which scored a 76% accuracy score. Similarly, after using AutoML, the generalized linear model (88%), gradient boosting model (87%), distributed random forest model (87%), extra tree model score (82%), and accuracy scores (82%), which ultimately mattered for model accuracy of prediction, were recommended for heart disease classification.
引用
收藏
页码:35025 / 35042
页数:18
相关论文
共 50 条
  • [1] Machine learning model matters its accuracy: a comparative study of ensemble learning and AutoML using heart disease prediction
    Yagyanath Rimal
    Siddhartha Paudel
    Navneet Sharma
    Abeer Alsadoon
    [J]. Multimedia Tools and Applications, 2024, 83 : 35025 - 35042
  • [2] Hyperparameter optimization: a comparative machine learning model analysis for enhanced heart disease prediction accuracy
    Rimal, Yagyanath
    Sharma, Navneet
    [J]. MULTIMEDIA TOOLS AND APPLICATIONS, 2023, 83 (18) : 55091 - 55107
  • [3] Hyperparameter optimization: a comparative machine learning model analysis for enhanced heart disease prediction accuracy
    Yagyanath Rimal
    Navneet Sharma
    [J]. Multimedia Tools and Applications, 2024, 83 : 55091 - 55107
  • [4] A Machine Learning Model for Predicting Heart Disease using Ensemble Methods
    Samagh J.S.
    Singh D.
    [J]. International Journal of Advanced Computer Science and Applications, 2022, 13 (09) : 558 - 565
  • [5] Heart Disease Prediction using Hybrid machine Learning Model
    Kavitha, M.
    Gnaneswar, G.
    Dinesh, R.
    Sai, Y. Rohith
    Suraj, R. Sai
    [J]. PROCEEDINGS OF THE 6TH INTERNATIONAL CONFERENCE ON INVENTIVE COMPUTATION TECHNOLOGIES (ICICT 2021), 2021, : 1329 - 1333
  • [6] Heart Disease Prediction Using Core Machine Learning Techniques-A Comparative Study
    Sarah, Sfurti
    Gourisaria, Mahendra Kumar
    Khare, Sandali
    Das, Himansu
    [J]. ADVANCES IN DATA AND INFORMATION SCIENCES, 2022, 318 : 247 - 260
  • [7] Study on Machine Learning based Heart Disease Prediction Model
    Zhang, Shihan
    [J]. PROCEEDINGS OF 2023 4TH INTERNATIONAL SYMPOSIUM ON ARTIFICIAL INTELLIGENCE FOR MEDICINE SCIENCE, ISAIMS 2023, 2023, : 346 - 352
  • [8] Liver disease prediction using machine learning and deep learning: A comparative study
    Singla, Bhawna
    Taneja, Soham
    Garg, Rishika
    Nagrath, Preeti
    [J]. INTELLIGENT DECISION TECHNOLOGIES-NETHERLANDS, 2022, 16 (01): : 71 - 84
  • [9] Machine Learning Techniques for Heart Disease Prediction: A Comparative Study and Analysis
    Katarya, Rahul
    Meena, Sunit Kumar
    [J]. HEALTH AND TECHNOLOGY, 2021, 11 (01) : 87 - 97
  • [10] Machine Learning Techniques for Heart Disease Prediction: A Comparative Study and Analysis
    Rahul Katarya
    Sunit Kumar Meena
    [J]. Health and Technology, 2021, 11 : 87 - 97