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

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.