Comparative analysis of linear regression, random forest, and LightGBM for hepatitis disease prediction

Hennie Tuhuteru, Goldy Valendria Nivaan, Marvelous Marvel Rijoly, Joselina Tuhuteru

Abstract


In bioinformatics research, computational pattern-analysis techniques are frequently employed to assist in disease prediction and diagnostic modeling, including applications for hepatitis prognosis. Hepatitis is a type of serious disease with various types that have the potential to threaten the life of the sufferer without showing significant symptoms and signs, so many sufferers do not realize that they are affected by the disease. Various methods are used to predict diseases in the hope of providing the best results from the learning model used. The objective of this study is to implement linear regression, random forest, and light gradient boosting machine (LightGBM) to estimate mortality risk among hepatitis patients. In addition, a performance comparison of the results of hepatitis disease prediction using the three algorithms was also carried out to find out which model gave the most accurate and optimal results. The results of this study show that the application of learning models from the linear regression, random forest and Light-GBM algorithms has been successfully carried out to predict the survival status of patients with hepatitis. The findings reveal that random forest achieved the highest predictive performance with an accuracy of 84%, followed by LightGBM at 77% and linear regression at 32%.

Keywords


Hepatitis; LightGBM; Linear regression; Random forest; Survival

Full Text:

PDF


DOI: http://doi.org/10.11591/ijeecs.v41.i1.pp430-438

Refbacks

  • There are currently no refbacks.


Creative Commons License
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.

Indonesian Journal of Electrical Engineering and Computer Science (IJEECS)
p-ISSN: 2502-4752, e-ISSN: 2502-4760
This journal is published by the Institute of Advanced Engineering and Science (IAES).

shopify stats IJEECS visitor statistics