Affiliation:
1. Sports and Health Management, Changzhou Vocational Institute of Engineering , Changzhou , Jiangsu , , China .
Abstract
Abstract
In this paper, a mental health evaluation system based on KB-SMOTE algorithm and XGBoost model was successfully established. Firstly, the questionnaire is reasonably set up and distributed to the college students majoring in physical education to fill in the questionnaire, to obtain the college students’ mental health questionnaire survey data. Secondly, the KB-SMOTE algorithm is used to optimize the data of questionnaire tuning data. The training set was utilized to establish the primary parameters of the XGBoost model. The final parameters were determined by the parameters that correspond to the highest accuracy of the XGBoost model. Finally, the model is built based on the best parameters obtained from the training set, and the optimized test data set is brought into the model to output the value of college students’ mental health assessment. The results show that the model performs best when the learning rate is 0.02, the tree depth is 5, the maximum number of iterations is 500, and the L2 regular term is 5. The XGBoost model predicts 4160 cases correctly and 286 cases incorrectly, with an overall correctness rate of 93.6%, and its corresponding ROC curve is closer to the y-axis, with an AUC value of 0.9154. By utilizing the XGBoost model, the model can accurately determine the mental health of college students. The mental health of college students can be accurately evaluated by models.