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

Journal: PloS one
Published Date:

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.

Authors

  • Yan Wei
    Zhejiang Key Laboratory for Agro-Food Processing, Zhejiang R &D Center for Food Technology and Equipment, Fuli Institute of Food Science, Zhejiang University, Hangzhou 310058, Zhejiang, China.
  • Xili Rao
    Department of Information Technology, Wenzhou Vocational College of Science and Technology, Wenzhou, 325006, China.
  • Yinjun Fu
    The Section of Employment, Wenzhou Vocational College of Science and Technology, Wenzhou, 325006, China.
  • Li Song
    Department of Obstetrics and Gynecology Qilu Hospital Cheeloo College of Medicine Shandong University Jinan Shandong China.
  • Huiling Chen
    College of Computer Science and Artificial Intelligence, Wenzhou University, Wenzhou 325035, China.
  • Junhong Li
    Luoyang Central Hospital affiliated to Zhengzhou University, Luoyang, China.