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 条
  • [41] Diabetes prediction in healthcare systems using machine learning algorithms on Hadoop cluster
    Yuvaraj, N.
    SriPreethaa, K. R.
    CLUSTER COMPUTING-THE JOURNAL OF NETWORKS SOFTWARE TOOLS AND APPLICATIONS, 2019, 22 (Suppl 1): : 1 - 9
  • [42] Analysis of Machine Learning Algorithms and Obtaining Highest Accuracy for Prediction of Diabetes in Women
    Agarwal, Arushi
    Saxena, Ankur
    PROCEEDINGS OF THE 2019 6TH INTERNATIONAL CONFERENCE ON COMPUTING FOR SUSTAINABLE GLOBAL DEVELOPMENT (INDIACOM), 2019, : 686 - 690
  • [43] Prediction of gestational diabetes mellitus in Asian women using machine learning algorithms
    Byung Soo Kang
    Seon Ui Lee
    Subeen Hong
    Sae Kyung Choi
    Jae Eun Shin
    Jeong Ha Wie
    Yun Sung Jo
    Yeon Hee Kim
    Kicheol Kil
    Yoo Hyun Chung
    Kyunghoon Jung
    Hanul Hong
    In Yang Park
    Hyun Sun Ko
    Scientific Reports, 13
  • [44] Diabetes prediction in healthcare systems using machine learning algorithms on Hadoop cluster
    N. Yuvaraj
    K. R. SriPreethaa
    Cluster Computing, 2019, 22 : 1 - 9
  • [45] Prediction of Arrhythmia with Machine Learning Algorithms
    Gursoy, Gunes
    Varol, Asaf
    9TH INTERNATIONAL SYMPOSIUM ON DIGITAL FORENSICS AND SECURITY (ISDFS'21), 2021,
  • [46] Empirical comparison of machine learning algorithms for bug prediction in open source software
    2017, Institute of Electrical and Electronics Engineers Inc., United States
  • [47] Machine learning algorithms for outcome prediction in (chemo)radiotherapy: An empirical comparison of classifiers
    Deist, Timo M.
    Dankers, Frank J. W. M.
    Valdes, Gilmer
    Wijsman, Robin
    Hsu, I-Chow
    Oberije, Cary
    Lustberg, Tim
    van Soest, Johan
    Hoebers, Frank
    Jochems, Arthur
    El Naqa, Issam
    Wee, Leonard
    Morin, Olivier
    Raleigh, David R.
    Bots, Wouter
    Kaanders, Johannes H.
    Belderbos, Jose
    Kwint, Margriet
    Solberg, Timothy
    Monshouwer, Rene
    Bussink, Johan
    Dekker, Andre
    Lambin, Philippe
    MEDICAL PHYSICS, 2018, 45 (07) : 3449 - 3459
  • [48] Comparison of Different Classification Algorithms for Prediction of Heart Disease by Machine Learning Techniques
    Harshitha B.
    Maria Rufina P.
    Shilpa B.L.
    SN Computer Science, 4 (2)
  • [50] Performance evaluation and comparison of machine learning algorithms for prediction of electrodeposited copper ions
    Deshmukh, Vimal Kumar
    Rajput, Mridul Singh
    Narang, H. K.
    ADVANCES IN MATERIALS AND PROCESSING TECHNOLOGIES, 2024, 10 (02) : 1282 - 1292