RESEARCH ARTICLE


A Study of Machine Learning Algorithms Performance Analysis in Disease Classification



Jai Kumar B1, *, Mohanasundaram R1
1 School of Computer Science and Engineering, Vellore Institute of Technology, Vellore, India


Article Metrics

CrossRef Citations:
0
Total Statistics:

Full-Text HTML Views: 224
Abstract HTML Views: 114
PDF Downloads: 115
ePub Downloads: 67
Total Views/Downloads: 520
Unique Statistics:

Full-Text HTML Views: 121
Abstract HTML Views: 73
PDF Downloads: 84
ePub Downloads: 54
Total Views/Downloads: 332



Creative Commons License
© 2024 The Author(s). Published by Bentham Open.

open-access license: This is an open access article distributed under the terms of the Creative Commons Attribution 4.0 International Public License (CC-BY 4.0), a copy of which is available at: https://creativecommons.org/licenses/by/4.0/legalcode. This license permits unrestricted use, distribution, and reproduction in any medium, provided the original author and source are credited.

* Address correspondence to this author at the School of Computer Science and Engineering, Vellore Institute of Technology, Vellore, India; E-mail: jaikumar.b2020a@vitstudent.ac.in


Abstract

Background

Because there are no symptoms, it might be difficult to detect CKD in its early stages. One of the main causes of CKD is diabetes mellitus (DM), and early detection of the condition can assist individuals in obtaining prompt treatment. Because this illness has no early signs and is only discovered after the kidneys have gone through 25% damage, early-stage prediction is not very likely. This is the key factor driving the need for early CKD prediction.

Objective

The objective of the paper is to find the best-performing learning algorithms that can be used to predict chronic kidney disease (CKD) at an earlier stage.

Methods

This research aimed to compare different machine learning algorithms used in different disease predictions by various researchers. In this comparative study, machine learning algorithms like Logistic Regression, K-Nearest Neighbor, Decision Tree, Support Vector Machine, Artificial Neural Network, Random Forest, Composite Hypercube on Iterated Random Projection, Naïve Bayes, J48, Ensembling, Multi-Layer Perceptron, Deep Neural Network, Autoencoder, and Long Short-Term Memory are used in disease classification.

Results

Each classification model is well tested in a different dataset, and out of these models, RF, DNN, and NB classification techniques give better performance in Diabetes and CKD prediction.

Conclusion

The RF, DNN, and NB classification algorithms worked well and achieved 100% accuracy in predicting diseases.

Keywords: Diabetes mellitus (DM), Chronic kidney disease (CKD), Random forest (RF), Naïve bayes (NB), Deep neural network (DNN), WHO.