Multi-Model Fusion Method and its Application in Prediction of Stock Index Movements

Yajie Zhang,Sijie Lu
DOI: https://doi.org/10.1145/3468891.3468900
2021-04-23
Abstract:Ensemble learning is a research direction in the field of machine learning. The combination of weak learners is enough to achieve good performance. TF-IDF method is used to represent the words distributed. It introduces the related machine learning algorithm theory, including the single classification algorithm of Logistic Regression, Random Forest, XGBoost algorithm, and compares the advantages and disadvantages of each algorithm. Three weak models are selected as the base classifier of stacking model fusion, and XGBoost is used as the secondary classifier of stacking model fusion. The accuracy and F1 score of prediction model achieves approximately 58%. Finally, the accuracy, generalization ability and stability of the learning system are significantly improved due to the multi-model fusion learning. It is helpful to reduce the risk of stock investment and improve the earning effect. The model fusion method based on Logistic Regression, Random Forest and XGBoost algorithm can also be applied to computer vision, medical diagnosis, biometric recognition, strategic games and other fields.
What problem does this paper attempt to address?