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

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

PLOS ONE, 2026, vol. 21, issue 9, 1-21

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.

Date: 2026
References: Add references at CitEc
Citations:

Downloads: (external link)
https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0359421 (text/html)
https://journals.plos.org/plosone/article/file?id= ... 59421&type=printable (application/pdf)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:plo:pone00:0359421

DOI: 10.1371/journal.pone.0359421

Access Statistics for this article

More articles in PLOS ONE from Public Library of Science
Bibliographic data for series maintained by plosone ().