Fine tuning attribute weighted naïve Bayes model for detecting anxiety disorder levels of online gamers

Anastasya Latubessy, Retantyo Wardoyo, Aina Musdholifah, Sri Kusrohmaniah

Abstract


This research applies the fine tuning attribute weighted naïve Bayes (FTAWNB) model using ordinal data. It is known that in previous research, the FTAWNB model outperformed its competitors on the dataset used. However, the FTAWNB model has not been applied in the mental health domain that uses ordinal data. Therefore, this research used the anxiety gamers dataset to test the fine-tuning attribute weighted Naïve Bayes (FTAWNB) model. Anxiety disorders are mental health disorders that can indicate the emergence of a gaming disorder. Gamers can experience anxiety disorders classified into four classes, namely minimal, mild, moderate, and severe anxiety. Then compare the results by FTAWNB obtained with three other naïve Bayes algorithms, namely Gaussian naïve Bayes, multinomial naïve Bayes, and categorical naïve Bayes, using the same dataset. Model performance is measured based on accuracy, precision, recall, and processing time. The test results show that the FTAWNB outperforms the other three models' accuracy, precision, and recall, with an accuracy value of 99.22%. While the accuracy of Gaussian NB is 91.132%, Categorical is 91.592%, and multinomial naïve Bayes is 61.104%. However, the FTAWNB takes slightly longer than the other three models' processing time. The FTAWNB takes 0.07 seconds to build the model and 0.05 seconds to test the model on training data.

Keywords


Anxiety disorder; Attribute weighted; Fine tuning; Naïve Bayes; Online gamer

Full Text:

PDF


DOI: http://doi.org/10.11591/ijece.v14i3.pp3277-3286

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

International Journal of Electrical and Computer Engineering (IJECE)
p-ISSN 2088-8708, e-ISSN 2722-2578

This journal is published by the Institute of Advanced Engineering and Science (IAES) in collaboration with Intelektual Pustaka Media Utama (IPMU).