TPE-optimized XGBoost model for fine-grained prediction of tertiary vocational college students' employment destinations.

Authors

Juntao Chen, Jinmei Zhan, Xiaodeng Zhou, Jiahua Yao, Su-Kit Tang, Xia Liu

Published in

PloS one. Volume 21. Issue 9. Pages e0359421. Epub Sep 28, 2026.

Abstract

Predicting tertiary vocational college students' employment destinations has become an urgent need for most tertiary vocational colleges. This study proposes a machine learning framework based on Tree-structured Parzen Estimator (TPE)-Optimized eXtreme gradient boosting (XGBoost) for fine-grained prediction of tertiary vocational college students' employment destinations. The dataset comprises 2,270 graduates (2016-2019 cohorts) from a single tertiary vocational college in China, characterized by 27 features spanning demographics, entrance scores, academic records, and extracurricular engagement. Employment destinations are categorized into three fine-grained classes: smooth employment, further study, and unemployment (class distribution: 78.8%, 17.8%, 3.4%). Recursive Feature Elimination (RFE) reduced the feature set to 22 informative variables. TPE hyperparameter optimization (100 iterations) was applied to the XGBoost classifier using an 80/20 train-test split, with stratified 5-fold cross-validation to mitigate overfitting risk. The proposed model achieves an overall accuracy of 81.72% and a balanced accuracy of 48.80% on the held-out test set, with macro-averaged precision of 0.6608, macro-averaged recall of 0.4880, macro-averaged specificity of 0.7659, and macro-averaged F1-score of 0.5263, outperforming seven baseline classifiers, including non-optimized XGBoost, logistic regression (LR), decision tree (DT), support vector machine (SVM), random forest (RF), K-nearest neighbor (KNN), and LightGBM. SHAP analysis identifies academic GPA across multiple semesters, Gaokao scores, and volunteer service hours as the most influential predictors.

PMID:
42804493
Bibliographic data and abstract were imported from PubMed on 29 Sep 2026.

Read full publication at:
Please sign in to see all details.

Stats

1-terrible, 9-excellent. How would you rate this publication? Sign in in to submit your rating.

Post a comment

You need to be signed in to post comments. You can sign in here.

Comments

There are no comments yet.