Next Article in Journal
Internet Traffic Prediction with Distributed Multi-Agent Learning
Previous Article in Journal
The Contribution of the User Experiences Goals for Designing Better Cobots: A Systematic Literature Review
 
 
Font Type:
Arial Georgia Verdana
Font Size:
Aa Aa Aa
Line Spacing:
Column Width:
Background:
Article

Prediction of Depression for Undergraduate Students Based on Imbalanced Data by Using Data Mining Techniques

by
Warawut Narkbunnum
and
Kittipol Wisaeng
*
Mahasarakham Business School, Mahasarakham University, Mahasarakham 44150, Thailand
*
Author to whom correspondence should be addressed.
Appl. Syst. Innov. 2022, 5(6), 120; https://doi.org/10.3390/asi5060120
Submission received: 2 November 2022 / Revised: 19 November 2022 / Accepted: 25 November 2022 / Published: 29 November 2022

Abstract

:
Depression is becoming one of the most prevalent mental disorders. This study looked at five different classification techniques to predict the risk of students’ depression based on their socio-demographics, internet addiction, alcohol use disorder, and stress levels to see if they were at risk for depression. We propose a combined sampling technique to improve the performance of the imbalanced classification of university student depression data. In addition, three different feature selection methods, Correlation, Gain ratio, and Relief feature selection algorithms, were used for extracting the most relevant features from the dataset. In our experimental results, we discovered that combining the bootstrapping technique with the Relief selection technique under sampling methods enabled the generation of a relatively well-balanced dataset on depression without significant loss of information. The results show that the overall accuracy in the risk of depression prediction data was 93.16%, outperforming the individual sampling technique. In addition, other evaluation metrics, including precision, recall, and area under the curve (AUC), were calculated for various models to determine the most effective model for predicting risk of depression.

1. Introduction

Depression is a common illness; more than 300 million people worldwide suffer from it. Depression can start at any time and affect anyone. It may occur once or on a recurring basis. The presence of persistent sadness and a loss of interest in normally enjoyable activities, along with an inability to perform daily tasks for at least two weeks, characterize depression. It is different from normal mood swings or short-term sadness caused by the problems of everyday life [1]. Furthermore, Thailand is experiencing an increase in the incidence of adolescent depression. Depression is 14.9% prevalent among Thai teenagers [2]. Depression has a direct detrimental impact on the quality of life of teenagers, including unpleasant emotions such as sadness, remorse, and low self-esteem. Data revealed that in 2017, the suicide rate among 20- to 24-year-olds was 4.94 per 100,000 people. In 2018, it increased to 5.33 per thousand people. According to yearly data, youth groups increasingly called the Mental Health Hotline 1323 to discuss mental health concerns. During 2018, 70,534 telephone consultations were provided, including 10,298 consultations with children aged 11 to 19 (14.6%) and 14,173 consultations with youth aged 20 to 25 (20.1%). Stress and anxiety are among the top five most prevalent issues among children and adolescents. Regarding psychiatric problems during the first six months of the fiscal year 2019, the service received 40,635 calls, including 13,635 calls from adolescents and young adults aged 11 to 25, according to Mahasarakham University students who visit mental health centers for treatment. During the academic years 2013 to 2017, students with major depressive disorders were diagnosed with 233 psychiatric disorders, or 36.5% [3].
The Depression Screening Question is a basic and time-consuming questionnaire that can be used for depression assessment and screening. If a patient has a depressed entrance score, an 8-question suicide assessment (8Q) is necessary to assess and continue managing the patient appropriately. Depression screening is, therefore, crucial and straightforward. It employs brief questions and 8Q and 9Q [4,5,6] tests to identify patients with depressive disorders and appropriately introduce them to the program. Although depression screening tools vary in their psychometric properties, the PHQ-9 is the most extensively psychometrically-tested tool [7]. There is a substantial number of standard questionnaires available in the Thai language, for example the Patient Health Questionnaire (2Q, 8Q, 9Q, and PHQ-A) [8,9,10] and the Beck Depression Inventory-Thai [11].
Knowledge of data mining is currently utilized in various fields of research, including medical research, as it examines techniques used to classify data in other domains to conduct research and utilize them to develop models for screening students at risk of depression [12,13,14]. Data mining in medicine is an emerging discipline of tremendous relevance for providing a prediction and a deeper understanding of illness classification, particularly in mental health domains concerning the most prevalent diseases of Alzheimer’s, schizophrenia, and depression. In terms of percentages, the main techniques applied to depression are support vector machines [15,16,17], followed by Naïve Bayes, Random Forest [18], logistic regression [12,19,20], and the Decision Tree [21]. It has been shown that data mining techniques are increasingly being used to predict depression in students [22,23,24,25]. Using data mining techniques and specific clinical and demographic factors that predict, we can identify depression risk in a timely and cost-effective manner. Some of these studies combine socio-demographic factors with physiological factors obtained from medical tests, which are also compatible with biomarker diagnosis [26,27,28].
In conclusion, data-mining techniques are accessible and effective diagnostic aids that we can include in the diagnostic workflow. Implementing them in public health care systems or by reaching out to the public, for example, by disseminating triggering advertisements offering to scan people’s social media and alert them if they need to seek help, can be excellent ways to alert people about a possible risk for depression [29].
The goal of this study was to determine whether a person is at risk of developing depression, to identify the key factors that cause depression, and to determine the best machine learning approach to identify at-risk people. This study also intended to minimize the required time for screening depression.
The major contributions to these goals are as follows:
  • Identifying the most important socio-demographic, internet addiction, alcohol use disorder, and stress factors of students that contribute to depression formation;
  • Creation of a dataset containing the students’ socio-demographics, internet addiction, alcohol use disorder, and stress to predict depression;
  • Exploring different data mining and feature selection algorithms to efficiently screen for the existence of depression;
  • Due to the simplicity of the required socio-demographic, internet addiction, alcohol usage disorder, and stress information of students used in this study, a student suspected to be at-risk for depression would feel less hassle giving the required information of this study to detect depression rather than answering the questions of different authentic depression screening scales.

2. Related Work

2.1. Depression

Depression is a mental health issue that affects people worldwide, ranking third after cardiac and respiratory diseases as a major cause of disability, particularly among university students [30]. In Thailand, one of Chiang Rai’s higher education schools was surveyed regarding the prevalence of depression among university students in years 1–4, ages 18 to 22. Using the Center for Epidemiological Studies-Depression Scale (CES-D), researchers discovered a prevalence of 31.9%. The greatest prevalence was in the Nursing Faculty [31]. Students in the first year of study at the Suranaree University of Technology responded to online questionnaires. The study revealed that 17.0% of students were depressed and 51.10% were stressed [32]. Among preclinical medical students, the University of Ubon Ratchathani found a 32.87% prevalence of depression considered mild, 27.97% were somewhat depressed, 4.20% were seriously depressed, and 0.70% enrolled in nursing school [33].
We should identify risk factors associated with depression among university students early in university to provide them with additional mental health support and prevent the worsening of risk factors.

2.2. Demographic Information

Many studies have established that the prevalence rate of depression varies from 2.8% to 10.3% based on socio-demographic factors [26]. The variables that have most often been studied in relation to mental health literacy are gender, college major, grade year, GPA, homeland [34], socio-economic factors [35], family status [36], health condition, and history of depression in the family [37].

2.3. Internet Addiction

In these modern times of digitization, the internet has become an integral part of everyday life, especially the lives of adolescents. Internet addiction is a severe problem with a profound impact on mental health [38,39,40]. The beginning of the twenty-first century has witnessed phenomenal growth in internet usage, particularly in developing nations such as Thailand [41]. As examined by this review, the impact of social media use on the incidence of depression, anxiety, and psychological distress among adolescents is likely to be multifactorial. The key findings of the included studies were categorized into four types of social media exposure: time spent, activity, investment, and addiction [40].
Internet addiction is an emerging concern, but there is a lack of comprehensive understanding of the causes of problematic behavior and of a gold-standard method for assessing symptoms. This study aimed to perform a psychometric analysis utilizing the most widely used screening instrument, the Young Internet Addiction Test (IAT) [42,43].

2.4. Alcohol Usage Disorder

Depression is characterized by fluctuating emotions. According to a literature review on psychosocial factors of depression associated with the risk of suicide in adolescents, several studies have been conducted in Thailand. Depressive disorders were found to be up to 14.3% of the factors associated with adolescent suicide, and reflections on the association between depression and suicide risk among secondary school-aged adolescents were conducted [44]. Depression was found to positively correlate with a statistically significant risk of suicide. The same holds for international research studies. In addition to concluding that it is important and necessary to study the correlation between depression and suicide risk, the review found that alcohol consumption was significantly statistically correlated with suicide risk [45,46]. Alcohol consumption is a major risk factor for university students [47] and depression is also positively associated with alcohol-related behaviors. Alcohol use disorder (AUD) and alcohol abuse are prevalent among adolescents and young adults and are linked to serious personal and social problems.
This study employed the Alcohol Use Disorders Identification Test (AUDIT), which is a simple and effective screening instrument for unhealthy alcohol use, which is defined as risky or harmful intake or any alcohol use disorder [48,49].

2.5. Stress

Daily stress is unavoidable. It is possible for anyone to experience it. It serves to improve awareness, promote learning, and increase productivity at work. Due to the need to adjust to rapid changes in physical, emotional, and social contexts, adolescence relates to an increased risk of mental health issues. Stress, anxiety, sadness, and suicide are all prevalent psychological issues in adolescents. Mental health difficulties affect adolescents in various ways, including the consequences of stress on self-efficacy. Students that are under a great deal of stress have low self-efficacy. Excessive stress can have a negative impact on both physical and mental health, self-esteem, and the success of learning and self-development. Teenagers are affected by mental health issues in a variety of ways, including the effects of stress on self-efficacy. One of the leading causes of suicide is depression. Financial restrictions, a lack of social connection, and concerns about transitioning to university life are connected with higher stress and anxiety among students [50].
This study, using a self-administered online questionnaire, attempted to quantify students’ stress levels and identify factors that influence depression. The stress measurement instrument was adapted from the Department of Mental Health, Ministry of Public Health’s stress evaluation form using the self-reported stress test (Suan Prung Stress Test, SPST-20) [51,52].

3. Materials and Methods

3.1. Instruments

The researchers followed the following procedures and methodologies:
  • Created a questionnaire based on the quantity of samples used for research and made sure the document was correct, complete, and had all the information it needed;
  • Asked for a formal letter from the university;
  • Collected information from each faculty; a process called “informed consent” was used to get the subjects’ consent and make sure they understand what is going on;
  • Held the processing period from June 2020 through March 2021;
  • Examined the responses to the questionnaire for completeness;
  • Collected information from the received questionnaires for further data analysis and interpretation.
The questionnaire consisted of five types, categorized as Type A–E, containing a total of 69 questions, and all questionnaires were non-identifiable to the participants. All questionnaires were filled out by the samplers and required between 20 and 45 min to complete. Type A–D consisted of 59 questions to be used as a predictive variable for depression risk, and Type E consisted of 9 questions to be used as a target variable. Each section’s questionnaire was as follows:

3.1.1. A Demographic Information Questionnaire (Type A)

In addition to relevant reviews, a personal information questionnaire that included questions about gender, college major, grade year, GPA, homeland, income adequacy, family status, health condition, and history of depression in the family.

3.1.2. Assessment of Internet Addiction (Type B)

The social network usage questionnaire adapted from the Young Internet Addiction Test was separated into three categories: frequency of use, length of use, and social network addiction. The test consisted of 12 questions using a Likert scale, which is used to determine the degree of agreement. The total IAT score was the sum of the examinee’s assessments for the 20-item responses. Each item was evaluated from 0 to 5 on a 5-point scale (from 1—not at all, to 5—always). The maximum possible score was 100. The severity of the problem increases as the score increases. Scores between 0 and 30 indicate Internet addiction, 50 to 79 show a moderate level, and 80 to 100 indicate a severe reliance on the Internet.

3.1.3. Alcohol Use Identification Test (AUDIT) (Type C)

The World Health Organization created this evaluation as a time- and space-constrained screening tool for excessive binge drinking. It can also assist in identifying problems with binge drinking that result in illnesses that send people to the doctor. The evaluation is a self-evaluation. It takes some time, but the evaluation is simple.
The AUDIT consists of 10 questions, each with a score of 0, 1, 2, 3, or 4, except for questions 9 and 10, which carry scores of 0, 2, and 4. The range of scores is from 0 to 40, with 0 indicating/showing an abstainer who has never experienced alcohol-related difficulties. The World Health Organization (WHO, Geneva, Switzerland) recommends a score of 1, and 7 indicates low-risk usage. Eight to fourteen points indicate potentially hazardous alcohol usage, while a score of fifteen or higher suggests the likelihood of alcohol dependence (moderate-severe alcohol use disorder).

3.1.4. Stress Test (Type D)

Thailand’s Department of Mental Health (2018) developed a stress test to evaluate what had happened over the previous six months. The SPST explores feelings over the previous 6 months, asking about occurrences and feelings towards such occurrences. The assessment criteria were as follows: 1 point of the stress score represents no stress, 2 points of the stress score represents low stress, 3 points of the stress score represents moderate stress, 4 points of the stress score represents high stress, and 5 points of the stress score represents the highest level of stress. Total scores were summed up and compared with the criteria for assessment of stress level as follows: 0–24 points indicate low stress, 25–42 points indicate moderate stress, 43–62 points indicate high stress, and more than 63 points indicates severe stress.

3.1.5. Patient Health Questionnaire for Adolescents (PHA-A)

A screening model called the Patient Health Questionnaire for Adolescents (PHQ-A) was created to evaluate depression in young people for the use of Thai teenagers. It is a brief examination, but adolescent depression assessments are accurate and congruent with research using earlier, original English-language instruments, including PHQ-9 and PHQ-A. A self-answer questionnaire with 9 items to gauge the severity of depressive symptoms, it is simple to use, reliable, and accurate when measuring depression in young people [53].

3.2. Data Set

The collection of data is a crucial component of model development. There are a limited number of data collection methods that are entirely dependent on the type of research being conducted. Observation, interview, document scanning, measurement, questionnaire, or a combination of these methods may be used to collect data [15]. In our work, data was collected from a questionnaire prepared by the authors.
According to the division of registration at Mahasarakham University, there were 37,579 undergraduate students registered for the second semester of the 2019 academic year. The sample size was calculated using Krejcie and Morgan’s pre-made table, which established the population’s proportion of desirable qualities, 0.5% acceptable tolerance levels, and 95% confidence intervals for 380 individuals.
The survey was conducted in the period between June 2020 and March 2021. The data set consisted of the responses of 460 participants. The inclusion criteria for this study were: either male or female, outpatients, aged more than 18 years, diagnosed without mood disorders, and volunteers who did not know they had depression. Participants were excluded if they were under the age of 18 years, illiterate, enrolled in other trials, or were in situations that did not allow obtaining written informed consent. Participants were not paid. The contents of the monitoring interviews were reviewed to identify patients who had attended at least 2 appointments. These criteria yielded the 380 participants
Among the 380 participants in the accumulated dataset, 296 participants were found to be at risk for depression. The prevalence of depression among the participants of the survey was 77.9%. Table 1 shows the distribution of depressed and non-depressed participants in the dataset.
Data in Table 1 are represented with imbalanced data, which is common when predicting with big data. Raw data are directly analyzed without using supplementary techniques such as a sample algorithm for data sets with imbalanced class ratios. This can decrease the performance of machine learning by causing prediction errors. Imbalanced data are data in which certain groups are overrepresented. One of the groups has a large amount of data, separated into majority and minority classes. Imbalanced information will result in the incorrect or less accurate categorization of the data of the minority group. However, it is possible to categorize the data of the majority group more precisely. General information can contain unbalanced information.
In order to improve the data before analyzing them, the Synthetic Minority Over-sampling Technique (SMOTE) [54,55,56,57] and resampling with replacement (bootstrapping) [58,59,60] were used in this study. With SMOTE, numerous variations exist for an oversampling technique that generates synthetic data by approximating minority instances. It is a type of oversampling method that has been demonstrated to be effective and is frequently utilized in machine learning with imbalanced high-dimensional data, and is increasingly utilized in the medical field. As an alternate way of assessing mediation, bootstrapping is a nonparametric resampling procedure that does not assume the normality of the sampling distribution. Bootstrapping is a computationally demanding method that estimates the indirect effect of each resampled data set by repeatedly sampling from the data set. In addition, as part of the preprocessing, the interviewer’s pauses and background noise were deleted.
This study prepared the data for sampling using the following methods: Table 2 shows the original data (D-ORI), the minority oversampling data (D-SM), and the resample with replacement data (D-RE).
From Table 2, it was determined that D-ORI had 380 instances, where majority or those at-risk were recorded for 296 cases and minority or those not at-risk were recorded for 84 cases, demonstrating that the imbalance between majority and minority equals 3.52. D-SM had 464 instances, with a majority of 296 and a minority of 168, for a ratio of 1.76, while D-RE had 380 instances, with a majority of 296 and a minority of 84, for a ratio of 3.52.

3.3. Target Variable

Informing the status of depression as screened by the PHQ-A questionnaire for measuring the target variable: if any of these variables’ value is true, the target variable is defined as “risk of depression” = 1, otherwise “non-risk of depression” = 0.

3.4. Feature Selection Techniques

During the construction of a model, only essential components should be chosen. The model’s performance may degrade if irrelevant features are selected. Selecting inappropriate characteristics may result in a deterioration of the model’s performance. Feature selection makes it easier to get rid of features that are not needed or are redundant and do not change how well the model works.
The following three strategies of characteristics evaluation were employed in the current study: CorrelationAttributeEval, GainRatioAttributeEval and ReliefFAttributeEval. The feature search strategy made use of the ranker method, and the following search techniques were implemented using the Weka software as described below:
CorrelationAttributeEval: This option can evaluate the worth of an attribute by measuring the correlation (Pearson’s) between it and the class;
GainRatioAttributeEval: This option can evaluate the worth of an attribute by measuring the gain ratio with respect to the class;
ReliefFAttributeEval: This option can evaluate the worth of an attribute by repeatedly sampling an instance and considering the value of the given attribute for the nearest instance of the same or different class that can operate on both discrete and continuous class data.

3.5. Data Mining Techniques for Depression Prediction

To predict depression, this study used 5 different data mining classifiers, namely: the Support Vector Machine (SVM), Naïve Bayes (NB), Logistic Regression (LR), Random Forest (RF), and Decision Tree (J48).

3.5.1. Support Vector Machine

SVM is one of the methods in the problem of pattern classification. SVM finds the optimal hyperplane that categorizes training data input into two classes (good and bad). Figure 1 shows the basic support vector machine.
The SVM algorithm is based on the kernel method, and the selection of the kernel type has a strong effect on the classification results. The type of kernel used was the Radial Basis Function (RBF) kernel, as in Equation (1):
K x , x = e x p γ ǁ x x ǁ 2 σ 2

3.5.2. Naïve Bayes

NB is a simple, supervised learning method for classification by calculating the probability to infer the solution. A conditional probability model is used as the data training model. It is appropriate for classifying a large sample, as calculated in Equation (2):
P A | B = P B | A , P A P B

3.5.3. Random Forest

RF is an ensemble learning technique. Multiple sets of training data and unique features are selected at random. The models are then constructed using a series of decision trees, where the out-of-bag data are collected for the prediction data test. Finally, the model outcomes are put to a vote, and the one that receives the most votes is the solution, as represented in Figure 2.

3.5.4. Logistic Regression

LR is applicable to data sets with dichotomous dependent and independent features. Due to the binary nature of the depression prediction data set, it cannot be modeled using linear regression. Logistic regression (LR) is necessary for such data. Two sets of experiments exist. One set produces a positive result, which is the attribute set by which depressive people are identified, while the other produces a negative result, which identifies those who are not depressed. The logistic function used in this model to predict the output of an experiment is illustrated in Equation (3).
f r = 1 1 + e r

3.5.5. Decision Tree

J48 is a classifier based on trees utilized for predictive modeling. It consists of an algorithm to construct decision tables and a visualization component to depict the model’s graphical user interface. It creates a hierarchical structure of the input data set based on the relationship among the data. The algorithm selects the most important attribute set from the original data set, reducing the likelihood of overestimation. From this small and compact output decision table, a classification decision is made about how to set rules for making decisions among certain qualities, as shown in Figure 3.
The configurations of the data mining classifier used in each experiment were determined in this study as shown in Table 3.

3.6. Training the Model

Following the selection of features, the models were built using the classification techniques. The 10-folds cross validation technique was used to validate the models’ performances. In this technique, the entire dataset is divided into 10 subsets and then processed 10 times. Nine subsets are used as testing sets, and the remaining one is used as a training set. Finally, the results are shown by averaging each of the 10 iterations. The subsets are divided using stratified sampling, meaning that each subset will have the same class ratio as the main dataset.

3.7. Performance Evaluation

Different performance metrics such as accuracy, precision, recall, and area under the curve (AUC) having been determined for all the models constructed in the previous subsection, the efficacies of these models were evaluated based on these performance metrics. Finally, the best model for predicting depression was chosen, according to the outcomes of these performance metrics.
The proposed model used WEKA and a confusion matrix to estimate classification algorithms: SVM, NB, RF, LR, and J48. For the model, there were three most common evaluation indexes: accuracy, precision, and recall. The calculation of these evaluation indexes is inseparable from the existence of a confusion matrix, which is often used in classification algorithms. Accuracy reflects the proportion of correct classification in the classification results of each category; that is, the accuracy of each category is judged by the model. The formula for this is provided in Equation (4). The ratio of accurately identified true positives to total positive samples is known as precision. The formula is provided in Equation (5). The recall rate reflects the sensitivity of the classification model to each category dataset. The recall percentages of correctly identified positive samples, total positive samples, and total false-negative samples are illustrated in Equation (6). The Area Under the Curve is a measure of a classifier’s ability to distinguish between classes and is used as a summary of the ROC curve. The higher the AUC, the better the model’s performance at distinguishing between the positive and negative classes, as shown in Table 4.
A c c u r a c y = t o t a l c o r r e c t p r e d i c t i o n t o t a l p r e d i c t i o n ,
P r e c i s i o n = T P T P + F P ,
R e c a l l = T P T P + F N ,
Our model employed classification techniques such as SVM, NB, RF, LR, and J48 for prediction. Figure 4 presents the proposed model structure.

4. Results

4.1. Data Analysis

Most of the respondents were female (88.95%, n = 338), in their 3rd year of study (51.05%, n = 194), with a GPA greater than 3.01 (32.89%, n = 125), from other provinces (84.47%, n = 321), with an adequate income (62.63%, n = 238), whose family was together (73.42%, n = 279), with no health condition (87.11%, n = 331), and no history of depression in the family (90.53%, n = 344). According to research on depression among college students in Thailand, most of the people who answer the surveys are female [32,61,62,63]

4.2. Internet Addiction Level, Alcohol Usage Identification Level, and Stress Level

The following can be described as the results of the questionnaire’s Parts B, C, and D: the degree of internet addiction was moderate for the majority at 75%, the level of alcohol use was predominantly low-risk at 66.3%, and the level of stress was predominantly high at 38.42% and severe at 33.68%

4.3. Evaluation

Table 5, Table 6 and Table 7 show the confusion matrices of the predicted results for each of the classifiers using a different feature selection technique. The measured accuracy, precision, recall, and area under the curve (AUC) of the classifiers for the constructed models are shown.
Table 5 represents the performance of the classification technique based on different balancing techniques. Without any balancing technique, the LR technique showed the best result, with an accuracy of 83.16%. With SMOTE, the RF technique showed the best result, with an accuracy of 82.97%. With bootstrapping, the RF technique showed the best result, with an accuracy of 90.53%.
Table 6 displays the performance of classifiers utilizing the Gain Ratio technique based on various balancing methods. Without any balancing technique, the J48 technique demonstrated the highest accuracy, at 83.68%. With SMOTE present, the RF and J48 techniques produced the best results, with an accuracy of 83.41%. With bootstrapping, the RF method achieved the best result, with an accuracy of 91.05%.
Table 7 represents the performance of the classification technique based on different balancing techniques. Without any balancing technique, the SVM technique showed the best result, with an accuracy of 84.47%. With SMOTE, the RF technique produced the best results, with an accuracy of 85.13%. With bootstrapping, the RF technique showed the best result, with an accuracy of 93.16%.

5. Discussion

Most of the publications included in our review presented high accuracy in classifying individuals with depression based on SMOTE techniques [55,64]. Balancing data by using the SMOTE technique is not the most accurate method for predicting risk of depression. Data balanced using the bootstrapping technique is more accurate than SMOTE.
In this study, we compared SMOTE and bootstrapping techniques. As a result, the accuracy of all the data mining classifiers dramatically increased by combining different feature selection techniques with a balancing technique, as shown in Figure 5. It was discovered that when using the SMOTE technique, the data mining technique’s accuracy value was reduced compared to when using the bootstrapping technique. Figure 5c demonstrates the highest degree of accuracy when using Relief selection techniques.
The purpose of this study was to identify and create the most significant predictor of depression in students, taking into account socio-demographics, internet addiction, alcohol use disorder, and stress factors, as shown in Table 8.

6. Conclusions

Different variables can contribute to the development of depression in a person. This study aimed at identifying the most prevalent risk factors for depression. To assess the risk of depression, a dataset of four types, as well as 380 participants, was compiled. Different feature selection techniques extracted the students’ most important socio-demographic, internet addiction, alcohol usage disorder, and stress factors responsible for forming the risk of depression. These feature selection techniques not only increased the training speed of the classifiers, but also improved their ability to accurately screen for depression. This study utilized five different data mining classifiers to determine the presence of depression. By observing the outcomes of the various models presented in this study, it was confirmed that the Random Forest classifier with the Relief feature selection technique was almost the perfect model to predict depression among the participants. It obtained an accuracy of 93.16%.
This work has only predicted the presence of depression in individual students. In the future, this study could be extended to identifying the severity of depression in students. As different socio-demographic and other factors have a big effect on how likely it is that a student will be depressed, this second study could look at the different characteristics of the participants. Several studies show that the performance of a model gets better when different data mining classifiers with balancing techniques and feature selection techniques are used during the data preprocessing steps. In the future, these methods could be used, their results could be compared to those of the current study, and this research could be developed into a more comprehensive depression assessment questionnaire for students than those currently in use.

Author Contributions

Conceptualization, W.N. and K.W.; data curation, W.N.; formal analysis, W.N. and K.W.; funding acquisition, W.N.; project administration, W.N. and K.W.; software analysis, W.N. and K.W.; supervision, W.N.; writing—original draft preparation, W.N. and K.W.; writing—review and editing, W.N. and K.W. All authors have read and agreed to the published version of the manuscript.

Funding

This research project was financially supported by Mahasarakham University (Grant No: 6603001-2566).

Informed Consent Statement

Not applicable.

Data Availability Statement

Not applicable.

Acknowledgments

This research project was financially supported by Mahasarakham University. The authors are also indebted to Hathaichanok Chompoopong for his English language consulting time and proofreading of the whole paper.

Conflicts of Interest

The authors declare no conflict of interest.

References

  1. WHO EMRO|What You Can Do|Mental Health. Available online: https://www.emro.who.int/mnh/what-you-can-do/index.html#accordionpan4 (accessed on 15 November 2022).
  2. Ritkumrop, K.; Surakarn, A.; Ekpanyaskul, C. The effectiveness of an integrated counseling program on emotional regulation among undergraduate students with depression. J. Health Res. 2021, 36, 186–198. [Google Scholar] [CrossRef]
  3. Jiranukool, J.; Kotputorn, N.; Sirisanokaeo, T. Prevalence of Psychiatric Disorders, Gender Differences, and a Comparison between Depressive or Anxiety Disorders and Adjustment Disorders among Mahasarakham University Students at Psychiatric Clinic. J. Psychiatr. Assoc. Thail. 2018, 63, 271–284. [Google Scholar]
  4. Dadfar, M.; Lester, D.; Hosseini, A.F.; Eslami, M. The Patient Health Questionnaire-9 (PHQ-9) as a brief screening tool for depression: A study of Iranian college students. Ment. Heal. Relig. Cult. 2021, 24, 850–861. [Google Scholar] [CrossRef]
  5. Li, D.; Chaudhary, H.; Zhang, Z. Modeling Spatiotemporal Pattern of Depressive Symptoms Caused by COVID-19 Using Social Media Data Mining. Int. J. Environ. Res. Public Health 2020, 17, 4988. [Google Scholar] [CrossRef]
  6. Gómez-Gómez, I.; Benítez, I.; Bellón, J.; Moreno-Peral, P.; Oliván-Blázquez, B.; Clavería, A.; Zabaleta-Del-Olmo, E.; Llobera, J.; Serrano-Ripoll, M.J.; Tamayo-Morales, O.; et al. Utility of PHQ-2, PHQ-8 and PHQ-9 for detecting major depression in primary health care: A validation study in Spain. Psychol. Med. 2022, 52, 1–11. [Google Scholar] [CrossRef] [PubMed]
  7. El-Den, S.; Chen, T.F.; Gan, Y.-L.; Wong, E.; O’Reilly, C.L. The psychometric properties of depression screening tools in primary healthcare settings: A systematic review. J. Affect. Disord. 2018, 225, 503–522. [Google Scholar] [CrossRef] [PubMed]
  8. Kawilapat, S.; Maneeton, B.; Maneeton, N.; Prasitwattanaseree, S.; Kongsuk, T.; Arunpongpaisal, S.; Leejongpermpoon, J.; Sukhawaha, S.; Traisathit, P. Comparison of unweighted and item response theory-based weighted sum scoring for the Nine-Questions Depression-Rating Scale in the Northern Thai Dialect. BMC Med. Res. Methodol. 2022, 22, 1–15. [Google Scholar] [CrossRef]
  9. Katchapakirin, K.; Wongpatikaseree, K.; Yomaboot, P.; Kaewpitakkun, Y. Facebook Social Media for Depression Detection in the Thai Community. In Proceedings of the 2018 15th International Joint Conference on Computer Science and Software Engineering (JCSSE), Nakhonpathom, Thailand, 11–13 July 2018; Institute of Electrical and Electronics Engineers Inc.: Piscataway, NJ, USA, 2018. [Google Scholar] [CrossRef]
  10. Panyawong, W.; Pavasuthipaisit, C.; Santitadakul, R. Thai version of the Patient Health Questionnaire for Adolescents in school setting. J. Ment. Health Thail. 2022, 30, 38–48. [Google Scholar]
  11. Sakunpong, N.; Ritkumrop, K. Psychometric Properties of the Thai Version Psychological Well-Being Scale and the Factors Related to among Thai Patients with Major Depressive Disorder. Depression Res. Treat. 2021, 2021, 1–7. [Google Scholar] [CrossRef] [PubMed]
  12. Alonso, S.G.; De La Torre-Díez, I.; Hamrioui, S.; López-Coronado, M.; Barreno, D.C.; Nozaleda, L.M.; Franco, M. Data Mining Algorithms and Techniques in Mental Health: A Systematic Review. J. Med. Syst. 2018, 42, 161. [Google Scholar] [CrossRef]
  13. Islam, S.; Hasan, M.; Wang, X.; Germack, H.D.; Alam, N.E. A Systematic Review on Healthcare Analytics: Application and Theoretical Perspective of Data Mining. Healthcare 2018, 6, 54. [Google Scholar] [CrossRef] [PubMed]
  14. Albahri, A.S.; Hamid, R.A.; Alwan, J.K.; Al-Qays, Z.; Zaidan, A.A.; Zaidan, B.B.; AlAmoodi, A.H.; Khlaf, J.M.; Almahdi, E.M.; Thabet, E.; et al. Role of biological Data Mining and Machine Learning Techniques in Detecting and Diagnosing the Novel Coronavirus (COVID-19): A Systematic Review. J. Med. Syst. 2020, 44, 1–11. [Google Scholar] [CrossRef] [PubMed]
  15. Srinath, K.S.; Kiran, K.; Pranavi, S.; Amrutha, M.; Shenoy, P.D.; Venugopal, K.R. Prediction of Depression, Anxiety and Stress Levels Using Dass-42. In Proceedings of the 2022 IEEE 7th International conference for Convergence in Technology (I2CT), Mumbai, India, 7–9 April 2022; pp. 1–6. [Google Scholar] [CrossRef]
  16. Ding, Y.; Chen, X.; Fu, Q.; Zhong, S. A Depression Recognition Method for College Students Using Deep Integrated Support Vector Algorithm. IEEE Access 2020, 8, 75616–75629. [Google Scholar] [CrossRef]
  17. Wijiasih, T.M.; Amriza, R.N.S.; Prabowo, D.A. The Classification of Anxiety, Depression, and Stress on Facebook Users Using the Support Vector Machine. JISA Jurnal Inform. dan Sains 2022, 5, 75–79. [Google Scholar] [CrossRef]
  18. Haque, U.M.; Kabir, E.; Khanam, R. Detection of child depression using machine learning methods. PLoS ONE 2021, 16, e0261131. [Google Scholar] [CrossRef] [PubMed]
  19. Aldarwish, M.M.; Ahmad, H.F. Predicting Depression Levels Using Social Media Posts. In Proceedings of the 2017 IEEE 13th International Symposium on Autonomous Decentralized System (ISADS), Bangkok, Thailand, 22–24 March 2017; pp. 277–280. [Google Scholar] [CrossRef]
  20. Choudhury, A.A.; Khan, R.H.; Nahim, N.Z.; Tulon, S.R.; Islam, S.; Chakrabarty, A. Predicting Depression in Bangladeshi Undergraduates using Machine Learning. In Proceedings of the 2019 IEEE Region 10 Symposium (TENSYMP), Kolkata, India, 7–9 June 2019; pp. 789–794. [Google Scholar] [CrossRef] [Green Version]
  21. Islam, R.; Kabir, M.A.; Ahmed, A.; Kamal, A.R.M.; Wang, H.; Ulhaq, A. Depression detection from social network data using machine learning techniques. Health Inf. Sci. Syst. 2018, 6, 8. [Google Scholar] [CrossRef]
  22. Hou, Y.; Xu, J.; Huang, Y.; Ma, X. A big data application to predict depression in the university based on the reading habits. In Proceedings of the 2016 3rd International Conference on Systems and Informatics (ICSAI), Shanghai, China, 19–21 November 2016; pp. 1085–1089. [Google Scholar] [CrossRef]
  23. Mohd, N.; Yahya, Y. A Data Mining Approach for Prediction of Students’ Depression Using Logistic Regression And Artificial Neural Network. In Proceedings of the 12th International Conference on Ubiquitous Information Management and Communication, Langkawi, Malaysia, 5–7 January 2018; p. 52. [Google Scholar] [CrossRef]
  24. Martinez-Martin, N.; Insel, T.R.; Dagum, P.; Greely, H.T.; Cho, M.K. Data mining for health: Staking out the ethical territory of digital phenotyping. NPJ Digit. Med. 2018, 1, 1–5. [Google Scholar] [CrossRef] [PubMed] [Green Version]
  25. Qu, D.; Guan, Q. Predictive Model and Analysis of Psychological Depression Based on College Students’ Behavior Data Mining. Wirel. Commun. Mob. Comput. 2022, 2022, 1–10. [Google Scholar] [CrossRef]
  26. Islam, R.; Adnan, R. Socio-Demographic Factors and Their Correlation with the Severity of Major Depressive Disorder: A Population Based Study. World J. Neurosci. 2017, 7, 193–202. [Google Scholar] [CrossRef] [Green Version]
  27. Tang, W.; Dai, Q. Depressive symptoms among first-year Chinese undergraduates: The roles of socio-demographics, coping style, and social support. Psychiatry Res. 2018, 270, 89–96. [Google Scholar] [CrossRef]
  28. Miles, R.; Rabin, L.; Krishnan, A.; Grandoit, E.; Kloskowski, K.; Miles, R.; Rabin, L.; Krishnan, A.; Grandoit, E.; Kloskowski, K. Mental health literacy in a diverse sample of undergraduate students: Demographic, psychological, and academic correlates. BMC Public Health 2020, 20, 1–13. [Google Scholar] [CrossRef] [PubMed]
  29. Richter, T.; Fishbain, B.; Richter-Levin, G.; Okon-Singer, H. Machine Learning-Based Behavioral Diagnostic Tools for Depression: Advances, Challenges, and Future Directions. J. Pers. Med. 2021, 11, 957. [Google Scholar] [CrossRef]
  30. Ibrahim, A.K.; Kelly, S.J.; Adams, C.E.; Glazebrook, C. A systematic review of studies of depression prevalence in university students. J. Psychiatr. Res. 2013, 47, 391–400. [Google Scholar] [CrossRef] [PubMed]
  31. Kaewkanta, P.; Rungreangkulkij, S. Prevalence of depression in higher education students in Chiangrai province. Nurs. J. C. 2013, 4, 48–64. [Google Scholar]
  32. Siripongpan, A.; Phattaramarut, K.; Namvichaisirikul, N.; Poochaya, S.; Horkaew, P. Prevalence of depression and stress among the first year students in Suranaree University of Technology, Thailand. Health Psychol. Res. 2022, 10, 35464. [Google Scholar] [CrossRef] [PubMed]
  33. Phanpanich, P.; Manwong, M.; Vongsuphakphan, P.; Udomsuk, L. Depression in the First to Third Year Medical Students of College of Medicine and Pubic Health, Ubonratchatani University. Srinagarind Med. J. 2021, 36, 287–294. [Google Scholar]
  34. Moeini, B.; Abasi, H.; Afshari, M.; Hosseini, M.H.; Ghaleiha, A. Homesickness, Depression and Happiness in University Students of Hamadan, Iran. Int. J. Epidemiol. Res. 2018, 5, 73–79. [Google Scholar] [CrossRef] [Green Version]
  35. Pabayo, R.; Dunn, E.C.; Gilman, S.; Kawachi, I.; Molnar, B.E. Income inequality within urban settings and depressive symptoms among adolescents. J. Epidemiol. Community Health 2016, 70, 997–1003. [Google Scholar] [CrossRef] [PubMed]
  36. Geshica, L.; Musabiq, S. Parents’ Marital Status and Psychological Distress among College Students. In Proceedings of the 2nd International Conference on Intervention and Applied Psychology (ICIAP 2018), Depok, West Java, Indonesia, 13–14 September 2018; pp. 366–373. [Google Scholar] [CrossRef] [Green Version]
  37. Ahmed, G.; Negash, A.; Kerebih, H.; Alemu, D.; Tesfaye, Y. Prevalence and associated factors of depression among Jimma University students. A cross-sectional study. Int. J. Ment. Health Syst. 2020, 14, 1–10. [Google Scholar] [CrossRef]
  38. Wahid, S.S.; Ottman, K.; Hudhud, R.; Gautam, K.; Fisher, H.L.; Kieling, C.; Mondelli, V.; Kohrt, B.A. Identifying risk factors and detection strategies for adolescent depression in diverse global settings: A Delphi consensus study. J. Affect. Disord. 2020, 279, 66–74. [Google Scholar] [CrossRef]
  39. Pantic, I.; Damjanovic, A.; Todorovic, J.; Topalovic, D.; Bojovic-Jovic, D.; Ristic, S.; Pantic, S. Association between online social networking and depression in high school students: Behavioral physiology viewpoint. Psychiatr. Danub. 2012, 24, 90–93. [Google Scholar] [PubMed]
  40. Keles, B.; McCrae, N.; Grealish, A. A systematic review: The influence of social media on depression, anxiety and psychological distress in adolescents. Int. J. Adolesc. Youth 2019, 25, 79–93. [Google Scholar] [CrossRef] [Green Version]
  41. Boonvisudhi, T.; Kuladee, S. Association between Internet addiction and depression in Thai medical students at Faculty of Medicine, Ramathibodi Hospital. PLoS ONE 2017, 12, e0174209. [Google Scholar] [CrossRef] [PubMed]
  42. Faraci, P.; Craparo, G.; Messina, R.; Severino, S. Internet Addiction Test (IAT): Which is the Best Factorial Solution? J. Med. Internet Res. 2013, 15, e225. [Google Scholar] [CrossRef] [PubMed]
  43. Samaha, A.A.; Fawaz, M.; El Yahfoufi, N.; Gebbawi, M.; Abdallah, H.; Baydoun, S.A.; Ghaddar, A.; Eid, A.H. Assessing the Psychometric Properties of the Internet Addiction Test (IAT) Among Lebanese College Students. Front. Public Health 2018, 6, 365. [Google Scholar] [CrossRef]
  44. Robkob, W.; Skulphan, S.; Sethabouppha, H. Depression, Alcohol Drinking Behaviors, and Suicidal Risks of Adolescents. Nurs. J. 2016, 45, 144–158. [Google Scholar]
  45. Briones, T.; Woods, J. Chronic binge-like alcohol consumption in adolescence causes depression-like symptoms possibly mediated by the effects of BDNF on neurogenesis. Neuroscience 2013, 254, 324–334. [Google Scholar] [CrossRef] [PubMed] [Green Version]
  46. Pedrelli, P.; Shapero, B.; Archibald, A.; Dale, C. Alcohol use and Depression During Adolescence and Young Adulthood: A Summary and Interpretation of Mixed Findings. Curr. Addict. Rep. 2016, 3, 91–97. [Google Scholar] [CrossRef] [PubMed]
  47. Rosenthal, S.R.; Clark, M.A.; Marshall, B.D.; Buka, S.L.; Carey, K.B.; Shepardson, R.L.; Carey, M.P. Alcohol consequences, not quantity, predict major depression onset among first-year female college students. Addict. Behav. 2018, 85, 70–76. [Google Scholar] [CrossRef] [Green Version]
  48. Pitanupong, J.; Ratanapinsiri, O. The Prevalence of Alcohol and Substance Use Among Medical Students at The Faculty of Medicine, Prince of Songkla University, 2016. J. Health Sci. Med. Res. 2018, 36, 107–115. [Google Scholar] [CrossRef] [Green Version]
  49. Phoosuwan, N. Prevalence and Factors Related to Alcohol Consumption among Undergraduate Students in a Northeast Thailand Province. Public Health J. Burapha Univ. 2019, 14, 1–15. [Google Scholar]
  50. Jenkins, E.K.; Slemon, A.; O’Flynn-Magee, K.; Mahy, J. Exploring the implications of a self-care assignment to foster undergraduate nursing student mental health: Findings from a survey research study. Nurse Educ. Today 2019, 81, 13–18. [Google Scholar] [CrossRef]
  51. Lueboonthavatchai, P.; Rodcharoen, N. Stress and associated factors in undergraduate students of Faculty of Law of a university in Bangkok. J. Med. Biosci. 2019, 1, 261–276. [Google Scholar] [CrossRef]
  52. Ounlam, T.; Panpoklang, S. Resilience Quotient and Level of Stress in Nursing Students at School of Nursing, Rangsit University. J. Health Nurs. Res. J. Boromarajonani Coll. Nurs. Bangk. 2021, 37, 240–251. [Google Scholar]
  53. Panyawong, W.; Pavasuthipaisit, C.; Santitadakul, R. Development of Thai Version of the Patient Health Questionnaire for Adolescents (PHQ-A); Child and Adolescent Mental Health Rajanagarindra Institute. Ministry of Public Health. 2017. Available online: https://dmh-elibrary.org/items/show/395 (accessed on 20 June 2022).
  54. Alghamdi, M.; Al-Mallah, M.; Keteyian, S.; Brawner, C.; Ehrman, J.; Sakr, S. Predicting diabetes mellitus using SMOTE and ensemble machine learning approach: The Henry Ford ExercIse Testing (FIT) project. PLoS ONE 2017, 12, e0179805. [Google Scholar] [CrossRef] [PubMed] [Green Version]
  55. Byeon, H. Predicting the Depression of the South Korean Elderly using SMOTE and an Imbalanced Binary Dataset. Int. J. Adv. Comput. Sci. Appl. 2021, 12. [Google Scholar] [CrossRef]
  56. Xin, L.K.; Rashid, N.B.A. Prediction of Depression among Women Using Random Oversampling and Random Forest. In Proceedings of the 2021 International Conference of Women in Data Science at Taif University (WiDSTaif), Taif, Saudi Arabia, 30–31 March 2021; pp. 1–5. [Google Scholar] [CrossRef]
  57. Aleem, S.; Huda, N.U.; Amin, R.; Khalid, S.; Alshamrani, S.S.; Alshehri, A. Machine Learning Algorithms for Depression: Diagnosis, Insights, and Research Directions. Electronics 2022, 11, 1111. [Google Scholar] [CrossRef]
  58. Abdallah_Qasaimeh, B.M.; Abdallah, A.; Ratte, S. Detecting Depression in Alzheimer and MCI Using Artificial Neural Networks (ANN). In Proceedings of the International Conference on Data Science, E-learning and Information Systems 2021, Ma’an, Jordan, 5–7 April 2021; pp. 250–253. [Google Scholar] [CrossRef]
  59. Taliaz, D.; Spinrad, A.; Barzilay, R.; Barnett-Itzhaki, Z.; Averbuch, D.; Teltsh, O.; Schurr, R.; Darki-Morag, S.; Lerer, B. Optimizing prediction of response to antidepressant medications using machine learning and integrated genetic, clinical, and demographic data. Transl. Psychiatry 2021, 11, 1–9. [Google Scholar] [CrossRef] [PubMed]
  60. Asare, K.O.; Moshe, I.; Terhorst, Y.; Vega, J.; Hosio, S.; Baumeister, H.; Pulkki-Råback, L.; Ferreira, D. Mood ratings and digital biomarkers from smartphone and wearable data differentiates and predicts depression status: A longitudinal data analysis. Pervasive Mob. Comput. 2022, 83, 101621. [Google Scholar] [CrossRef]
  61. Angkapanichkit, J.; Rochanahastin, A. Depression Situation for Thai University Students. Suranaree J. Soc. Sci. 2021, 15, 61–86. [Google Scholar] [CrossRef]
  62. Singh, R. An Assessment of Depression, Anxiety, and their Associated Factors amongst Doctoral Students in Bangkok, Thailand. Int. J. Multidiscip. Res. Publ. 2020, 3, 9–18. [Google Scholar]
  63. Rhein, D.; Nanni, A. Assessing Mental Health Among Thai University Students: A Cross-Sectional Study. SAGE Open 2022, 12, 21582440221129248. [Google Scholar] [CrossRef]
  64. Jakobsen, P.; Garcia-Ceja, E.; Riegler, M.; Stabell, L.A.; Nordgreen, T.; Torresen, J.; Fasmer, O.B.; Oedegaard, K.J. Applying machine learning in motor activity time series of depressed bipolar and unipolar patients compared to healthy controls. PLoS ONE 2020, 15, e0231995. [Google Scholar] [CrossRef]
Figure 1. Base of support vector machines.
Figure 1. Base of support vector machines.
Asi 05 00120 g001
Figure 2. Typical Random Forest.
Figure 2. Typical Random Forest.
Asi 05 00120 g002
Figure 3. Classifier tree visualizer. Tree J48 from WEKA.
Figure 3. Classifier tree visualizer. Tree J48 from WEKA.
Asi 05 00120 g003
Figure 4. The flow diagram of the proposed model.
Figure 4. The flow diagram of the proposed model.
Asi 05 00120 g004
Figure 5. Accuracy of the classifiers for different feature selection techniques: (a) Comparing outcomes with the correlation method; (b) Comparing outcomes with the gain ratio method; and (c) Comparing outcomes with the relief method.
Figure 5. Accuracy of the classifiers for different feature selection techniques: (a) Comparing outcomes with the correlation method; (b) Comparing outcomes with the gain ratio method; and (c) Comparing outcomes with the relief method.
Asi 05 00120 g005
Table 1. Distribution of depressed and non-depressed dataset participants.
Table 1. Distribution of depressed and non-depressed dataset participants.
ClassNumber of ParticipantsPercentage (%)
Depressed29677.9
Non-depressed8422.1
Table 2. Research data set after data preparation.
Table 2. Research data set after data preparation.
Data InputInstancesMajority (Yes)Minority (No)Imbalance Ratio (maj/min)
D-ORI380296843.52
D-SM4642961681.76
D-RE380296843.52
Table 3. Parameters of the data mining techniques.
Table 3. Parameters of the data mining techniques.
TechniqueParameter Setting
SVMBatch size = 100, cache size = 40, kernel type = RBF, gamma = 0
NBBatch size = 100, numDecimalPlaces = 2
LRBatch size = 100, numDecimalPlaces = 4
RFBatch size = 100, numDecimalPlaces = 2, numExecutionSlots = 1, seed = 1
J48Batch size = 100, confidence factor = 0.25, mumDecimalPlaces = 2, seed = 1
Table 4. The expert scale for AUC values shows the quality of the test.
Table 4. The expert scale for AUC values shows the quality of the test.
AUC ValueTest Quality
0.90–1.00Excellent
0.80–0.90Very good
0.70–0.80Good
0.60–0.70Satisfactory
0.50–0.60Unsatisfactory
Table 5. Performance of the classifiers using the correlation technique.
Table 5. Performance of the classifiers using the correlation technique.
Balancing TechniqueClassifier TechniqueAccuracy (%)Precision (%)Recall (%)AUC
Without BalancingSVM82.370.8460.9460.669
NB76.840.9130.7770.841
LR83.160.8650.9290.838
RF79.210.8470.8950.782
J4878.950.8530.8820.709
SMOTESVM79.740.8240.8680.770
NB75.650.8730.7230.859
LR78.020.8300.8240.874
RF82.970.8510.8890.889
J4878.020.8340.8180.784
BootstrappingSVM83.160.8520.9490.683
NB77.630.9140.7870.861
LR81.840.8600.9160.863
RF90.530.9280.9530.946
J4886.320.9010.9260.830
Table 6. Performance of the classifiers using the Gain Ratio technique.
Table 6. Performance of the classifiers using the Gain Ratio technique.
Balancing TechniqueClassifier TechniqueAccuracy (%)Precision (%)Recall (%)AUC
Without BalancingSVM83.680.8520.9560.901
NB76.840.9130.7770.839
LR83.160.8650.9290.896
RF82.110.8770.8950.886
J4881.840.8700.9020.886
SMOTESVM81.250.8340.8820.857
NB76.720.8920.7230.799
LR79.560.8270.8580.842
RF83.410.8710.8680.870
J4883.410.8480.9020.874
BootstrappingSVM85.790.8560.9830.915
NB80.790.9370.8070.868
LR83.420.8610.9390.898
RF91.050.9460.9390.942
J4887.110.9020.9360.919
Table 7. Performance of the classifiers using the Relief technique.
Table 7. Performance of the classifiers using the Relief technique.
Balancing TechniqueClassifier TechniqueAccuracy (%)Precision (%)Recall (%)AUC
Without BalancingSVM84.470.8580.9590.906
NB75.000.8970.7670.827
LR83.950.8710.9320.900
RF82.630.8640.9220.892
J4881.580.8670.9020.884
SMOTESVM75.220.8550.7360.791
NB82.330.8500.8780.864
LR78.230.8420.8110.826
RF85.130.8540.9260.888
J4879.740.8220.8720.846
BootstrappingSVM85.530.8550.9800.913
NB75.260.9010.7670.828
LR84.740.8720.9430.906
RF93.160.9440.9700.957
J48 0.8970.9390.917
Table 8. Selected attributes involved in each type by using the Relief technique.
Table 8. Selected attributes involved in each type by using the Relief technique.
AttributesDescription
Demographic A5Homeland
A6Income adequacy
A7Family status
Internet AddictionB1How often do you find that you stay online longer than you intended?
B14How often do you form new relationships with fellow online users?
B19How often do you choose to spend more time online over going out with others?
Alcohol Usage C3How often do you have 5 or more drinks on one occasion?
C9Have you or someone else been injured as a result of your drinking?
C10Has a relative, a friend, a doctor, or another health worker been concerned about your drinking or suggested you cut down?
StressD1I have been afraid of making mistakes.
D15I have felt sad and depressed.
D19I have often felt fatigued.
Publisher’s Note: MDPI stays neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Share and Cite

MDPI and ACS Style

Narkbunnum, W.; Wisaeng, K. Prediction of Depression for Undergraduate Students Based on Imbalanced Data by Using Data Mining Techniques. Appl. Syst. Innov. 2022, 5, 120. https://doi.org/10.3390/asi5060120

AMA Style

Narkbunnum W, Wisaeng K. Prediction of Depression for Undergraduate Students Based on Imbalanced Data by Using Data Mining Techniques. Applied System Innovation. 2022; 5(6):120. https://doi.org/10.3390/asi5060120

Chicago/Turabian Style

Narkbunnum, Warawut, and Kittipol Wisaeng. 2022. "Prediction of Depression for Undergraduate Students Based on Imbalanced Data by Using Data Mining Techniques" Applied System Innovation 5, no. 6: 120. https://doi.org/10.3390/asi5060120

Article Metrics

Back to TopTop