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 条
  • [21] DIABETES PREDICTION MODEL AND DIAGNOSTIC SYSTEM BASED ON MACHINE LEARNING ALGORITHMS
    Yu, H. P.
    Li, F. Y.
    Xie, Y. Q.
    Guo, M.
    BASIC & CLINICAL PHARMACOLOGY & TOXICOLOGY, 2017, 121 : 53 - 53
  • [22] Early Prediction of Diabetes Using Feature Selection and Machine Learning Algorithms
    Abdollahi J.
    Aref S.
    SN Computer Science, 5 (2)
  • [23] An Extensive Survey on Recent Machine Learning Algorithms for Diabetes Mellitus Prediction
    Selvi, R. Thanga
    Muthulakshmi, I
    INTELLIGENT COMMUNICATION TECHNOLOGIES AND VIRTUAL MOBILE NETWORKS, ICICV 2019, 2020, 33 : 328 - 335
  • [24] IoT Technologies and Machine Learning Algorithms on Diabetes Prediction-Survey
    Thilagavathi, G.
    Karthikeyan, N. K.
    Priyadharshini, G.
    Lavanya, G.
    PROCEEDINGS OF SECOND INTERNATIONAL CONFERENCE ON SUSTAINABLE EXPERT SYSTEMS (ICSES 2021), 2022, 351 : 869 - 876
  • [25] A Comparative Study with Different Machine Learning Algorithms for Diabetes Disease Prediction
    Kibria, Hafsa Binte
    Matin, Abdul
    Jahan, Nusrat
    Islam, Sanzida
    2021 18TH INTERNATIONAL CONFERENCE ON ELECTRICAL ENGINEERING, COMPUTING SCIENCE AND AUTOMATIC CONTROL (CCE 2021), 2021,
  • [26] Diabetes prediction using feature engineering and machine learning algorithms with security
    Arora, Jyoti
    Rathee, Sonia
    Gahlan, Mamta
    Shalu, Amita Yadav
    JOURNAL OF STATISTICS AND MANAGEMENT SYSTEMS, 2024, 27 (02) : 273 - 284
  • [27] Comparison of Machine Learning Algorithms for Bug Prediction in Open Source Software
    Malhotra, Ruchika
    Bahl, Laavanye
    Sehgal, Sushant
    Priya, Pragati
    PROCEEDINGS OF THE 2017 INTERNATIONAL CONFERENCE ON BIG DATA ANALYTICS AND COMPUTATIONAL INTELLIGENCE (ICBDAC), 2017, : 40 - 45
  • [28] Prediction of progression rate and fate of osteoarthritis: Comparison of machine learning algorithms
    Yoo, Hyun Jin
    Jeong, Ho Won
    Kim, Sung Woon
    Kim, Myeongju
    Lee, Jae Ik
    Lee, Yong Seuk
    JOURNAL OF ORTHOPAEDIC RESEARCH, 2023, 41 (03) : 583 - 590
  • [29] Comparison of Machine Learning Algorithms for Creation of a Bitcoin Price Prediction Model
    Cibula, Milan
    Tkac, Michal
    POLITICKA EKONOMIE, 2023, 71 (05) : 496 - 517
  • [30] Comparison of the prediction accuracy of machine learning algorithms in crosslinguistic vowel classification
    Georgiou, Georgios P.
    SCIENTIFIC REPORTS, 2023, 13 (01):