A comparison of machine learning algorithms for diabetes prediction

被引:62
|
作者
Khanam, Jobeda Jamal [1 ]
Foo, Simon Y. [1 ]
机构
[1] FAMU FSU Coll Engn, Dept Elect & Comp Engn, Tallahassee, FL 32310 USA
来源
ICT EXPRESS | 2021年 / 7卷 / 04期
关键词
Machine learning; Data Mining; Neural Network; K-fold Cross Validation; Accuracy;
D O I
10.1016/j.icte.2021.02.004
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Diabetes is a disease that has no permanent cure; hence early detection is required. Data mining, machine learning (ML) algorithms, and Neural Network (NN) methods are used in diabetes prediction in our research. We used the Pima Indian Diabetes (PID) dataset for our research, collected from the UCI Machine Learning Repository. The dataset contains information about 768 patients and their corresponding nine unique attributes. We used seven ML algorithms on the dataset to predict diabetes. We found that the model with Logistic Regression (LR) and Support Vector Machine (SVM) works well on diabetes prediction. We built the NN model with a different hidden layer with various epochs and observed the NN with two hidden layers provided 88.6% accuracy. (C) 2021 The Korean Institute of Communications and Information Sciences (KICS). Publishing services by Elsevier B.V.
引用
收藏
页码:432 / 439
页数:8
相关论文
共 50 条
  • [1] Comparison of Machine Learning Algorithms for Prediction of Diabetes
    Costea, Naomi Estera
    Moisi, Elisa Valentina
    Popescu, Daniela Elena
    2021 16TH INTERNATIONAL CONFERENCE ON ENGINEERING OF MODERN ELECTRIC SYSTEMS (EMES), 2021, : 56 - 59
  • [2] Diabetes Prediction using Machine Learning Algorithms
    Mujumdar, Aishwarya
    Vaidehi, V.
    2ND INTERNATIONAL CONFERENCE ON RECENT TRENDS IN ADVANCED COMPUTING ICRTAC -DISRUP - TIV INNOVATION , 2019, 2019, 165 : 292 - 299
  • [3] Prediction of Gestational Diabetes by Machine Learning Algorithms
    Gnanadass I.
    IEEE Potentials, 2020, 39 (06): : 32 - 37
  • [4] Prediction of Diabetes Using Machine Learning Algorithms in Healthcare
    Sarwar, Muhammad Azeem
    Kamal, Nasir
    Hamid, Wajeeha
    Shah, Munam Ali
    2018 24TH IEEE INTERNATIONAL CONFERENCE ON AUTOMATION AND COMPUTING (ICAC' 18), 2018, : 247 - 252
  • [5] Diabetes Prediction Using Machine Learning Algorithms and Ontology
    El Massari H.
    Sabouri Z.
    Mhammedi S.
    Gherabi N.
    Journal of ICT Standardization, 2022, 10 (02): : 319 - 338
  • [6] Diabetes Disease Prediction Using Machine Learning Algorithms
    Lyngdoh, Arwatki Chen
    Choudhury, Nurul Amin
    Moulik, Soumen
    2020 IEEE-EMBS CONFERENCE ON BIOMEDICAL ENGINEERING AND SCIENCES (IECBES 2020): LEADING MODERN HEALTHCARE TECHNOLOGY ENHANCING WELLNESS, 2021, : 517 - 521
  • [7] Comparison of Machine Learning Algorithms in Restaurant Revenue Prediction
    Gogolev, Stepan
    Ozhegov, Evgeniy M.
    ANALYSIS OF IMAGES, SOCIAL NETWORKS AND TEXTS (AIST 2019), 2020, 1086 : 27 - 36
  • [8] Comparison of Machine Learning Algorithms for Crime Prediction in Dubai
    Alabdouli, Shaikha Khamis
    Alomosh, Ahmad Falah
    Nassif, Ali Bou
    Nasir, Qassim
    INTERNATIONAL JOURNAL OF ADVANCED COMPUTER SCIENCE AND APPLICATIONS, 2023, 14 (09) : 169 - 173
  • [9] Application of machine learning algorithms in early prediction of diabetes dataset
    Dou, Yifeng
    ASIA-PACIFIC JOURNAL OF CLINICAL ONCOLOGY, 2022, 18 : 46 - 47
  • [10] Performance Comparison of Machine Learning Models for Diabetes Prediction
    Cihan, Pinar
    Coskun, Hakan
    29TH IEEE CONFERENCE ON SIGNAL PROCESSING AND COMMUNICATIONS APPLICATIONS (SIU 2021), 2021,