Machine learning prediction model based on enhanced bat algorithm and support vector machine for slow employment prediction

PLoS One. 2023 Nov 9;18(11):e0294114. doi: 10.1371/journal.pone.0294114. eCollection 2023.

Abstract

The employment of college students is an important issue that affects national development and social stability. In recent years, the increase in the number of graduates, the pressure of employment, and the epidemic have made the phenomenon of 'slow employment' increasingly prominent, becoming an urgent problem to be solved. Data mining and machine learning methods are used to analyze and predict the employment prospects for graduates and provide effective employment guidance and services for universities, governments, and graduates. It is a feasible solution to alleviate the problem of 'slow employment' of graduates. Therefore, this study proposed a feature selection prediction model (bGEBA-SVM) based on an improved bat algorithm and support vector machine by extracting 1694 college graduates from 2022 classes in Zhejiang Province. To improve the search efficiency and accuracy of the optimal feature subset, this paper proposed an enhanced bat algorithm based on the Gaussian distribution-based and elimination strategies for optimizing the feature set. The training data were input to the support vector machine for prediction. The proposed method is experimented by comparing it with peers, well-known machine learning models on the IEEE CEC2017 benchmark functions, public datasets, and graduate employment prediction dataset. The experimental results show that bGEBA-SVM can obtain higher prediction Accuracy, which can reach 93.86%. In addition, further education, student leader experience, family situation, career planning, and employment structure are more relevant characteristics that affect employment outcomes. In summary, bGEBA-SVM can be regarded as an employment prediction model with strong performance and high interpretability.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Algorithms*
  • Employment
  • Humans
  • Machine Learning
  • Support Vector Machine*

Grants and funding

This work was supported in part by the Philosophy and Social Science Foundation in Zhejiang Province of China under Grant 23GXSZ68YB, the Humanities and Social Sciences Program of the Ministry of Education under Grant 23JDSZ3236, the Scientific and Technological Innovation Project of Zhejiang University Students under Grant 2023R253002 and the Philosophy and Social Science Project of Wenzhou, China, under Grant 23BM038YB. The four projects were carried out by the primary author of this paper Yan Wei. These projects primarily focused on the research of innovation, entrepreneurship, and employment education, providing a solid research foundation and data support for the writing of this paper. Consequently, this facilitated the organization of ideas and propelled the process of data analysis.