Stock Trend Prediction and Analysis Based on Machine Learning

Authors

  • Ailin Sun

DOI:

https://doi.org/10.54097/vbx90z77

Keywords:

Stock trend prediction, Support Vector Machines (SVM), Random Forest, eXtreme Gradient Boosting (XGBoost), Principal Component Analysis (PCA).

Abstract

Stock trend prediction is a critically important activity in the financial sector. Investors aim to gain a better understanding of market dynamics, make wiser investment decisions, and manage investment risks more effectively by forecasting stock price trends. This is the focus of our upcoming research. This paper aims to utilize machine learning methods such as Support Vector Machines (SVM), based on spatial interval maximization strategies, Random Forest, based on multi-decision tree parallel learning strategies, and eXtreme Gradient Boosting (XGBoost), based on classification functions, to devise accurate classification function strategies for predicting and analyzing fluctuations in Dow Jones stock prices. Principal Component Analysis (PCA) will then be used to select some suitable features, identifying key factors with the greatest impact and minimal inter-correlation related to stock prices. Through a combination of various classification models, the research aims to forecast the future rise and fall of Dow Jones stocks. It has been observed that performing feature curation through PCA in advance enhances the predictive accuracy of different machine learning methods. Notably, the combination of PCA and XGBoost achieves the highest predictive rate. This approach provides a robust and scientifically grounded method for future stock price predictions and the formulation of trading strategies.

Downloads

Download data is not yet available.

References

Abu-Mostafa, Y.S., Atiya, A.F. “Introduction to financial forecasting”. Applied Intelligence. 1996. 205 – 213.

Hyun jung Kim, Kyung shik Shin. “A hybrid approach based on neural networks and genetic algorithms for detecting temporal patterns in stock markets”. Applied Soft Computing Journal. 2007, 569 – 576.

Erkam Guresen, Gulgun Kayakutlu, and Tugrul U. Daim.“Using artificial neural network models in stock market index prediction”. Expert Systems with Applications. 2011, 10389 – 10397.

Yuling Lin, Haixiang Guo, and Jinglu Hu.“An SVM-based approach for stock market trend prediction”. The 2013 International Joint Conference on Neural Networks (IJCNN). 2013, 1 – 7.

Ash Booth, Enrico Gerding, and Frank McGroarty. “Predicting equity market price impact with performance weighted ensembles of random forests”. 2014 IEEE Conference on Computational Intelligence for Financial Engineering Economics (CIFEr). 2014, 286 – 293.

Tianqi Chen, Carlos Guestrin. “XGBoost: A Scalable Tree Boosting System”. Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining. KDD ’16. ACM. 2016.

Kyung Keun Yun, Sang Won Yoon, and Daehan Won. “Prediction of stock price direction using a hybrid GA-XGBoost algorithm with a three-stage feature engineering process”. Expert Systems with Applications. 2021.

Sadegh Jalalian. https://www.kaggle.com/sadeghjalalian.

Downloads

Published

10-04-2024

How to Cite

Sun, A. (2024). Stock Trend Prediction and Analysis Based on Machine Learning. Highlights in Science, Engineering and Technology, 92, 165-170. https://doi.org/10.54097/vbx90z77