Scientific Journal of Informatics
Vol. 11 No. 3: August 2024

Hyperparameter Tuning Decision Tree and Recursive Feature Elimination Technique for Improved Chronic Kidney Disease Classification

Saputra, Aries Gilang (Unknown)
Purwanto, Purwanto (Unknown)
Pujiono, Pujiono (Unknown)



Article Info

Publish Date
06 Nov 2024

Abstract

Purpose: This study has the purpose of classifying patients with chronic kidney disease based on specific features and improving the classification models by tuning hyperparameters. This study aims to detect chronic kidney disease at an early stage. Methods: In this study, a machine learning classifier in the form of a decision tree is used to classify chronic kidney disease on the Risk Factor Prediction of Chronic Kidney Disease dataset. After that, the performance of the classifier model is improved by using feature selection, namely Recursive Feature Elimination and Hyperparameter tuning with GridSearchCV. Result: After tests were conducted 3 times namely testing with Decision Tree, Recursive Feature Elimination, and Hyperparameter tuning GridSearchCV which is the proposed method, then compared to other tests. The results from this study is using that method can improve the Decision Tree classifier in classifying chronic kidney disease patients. Novelty: Dataset that have been used in this study is from UCI machine learning repository namely Risk Factor Prediction of Chronic Kidney Disease that have 202 instances and 28 feature and after being processess and conducting test, Recursive Feature Elimination and Hyperparameter tuning GridSearchCV can improve the Decision Tree classifier in classifying chronic kidney disease.

Copyrights © 2024






Journal Info

Abbrev

sji

Publisher

Subject

Computer Science & IT Control & Systems Engineering Decision Sciences, Operations Research & Management Electrical & Electronics Engineering Engineering

Description

Scientific Journal of Informatics (p-ISSN 2407-7658 | e-ISSN 2460-0040) published by the Department of Computer Science, Universitas Negeri Semarang, a scientific journal of Information Systems and Information Technology which includes scholarly writings on pure research and applied research in the ...