Machine Learning Algorithm for Stock Market Prediction – A Comparison
PDF

Keywords

Stock Exchange
Yahoo Finance
DT
LSTM
Machine Learning Accuracy

How to Cite

Gokul, A., R. Manoj Kumar, and A. Jayasurya. 2023. “Machine Learning Algorithm for Stock Market Prediction – A Comparison”. Journal of Artificial Intelligence and Capsule Networks 5 (4): 521-33. https://doi.org/10.36548/jaicn.2023.4.005.

Abstract

The prediction about the stock market serves as an effort to forecast the value of the market, an individual stock, or a particular industrial sector. The prediction or forecast is usually done by using several approaches and analyzing the fundamental or technical details of an industry, an economy, or both. Predicting stock markets is very essential, as successful prediction can help in proper decision-making as well as in increasing the profit of the business. As prediction of the stock market is a bit complicated and challenging, conventional methods do not consistently forecast the changes with absolute certainty. For this reason, the proposed study has come up with a comparison of the machine learning models for forecasting the market changes. The aim of the study is to compare two machine learning models: the decision tree (DT) algorithm, which excels at handling nonlinear relationships and feature interaction, and the long short-term memory (LSTM) algorithm, which could efficiently capture long-range dependencies in sequential data. The Yahoo Finance dataset was used for performing the comparison. The results observed will be utilized to analyze the stock expenses and their predictions in the future.

PDF

References

Box GEP, Jenkins GM, Reinsel GC, Ljung GM. Time series analysis: forecasting and control. New York: Wiley; 2015.

Huang, Chao, Li-li Huang, and Ting-ting Han. "Financial time series forecasting based on wavelet kernel support vector machine." In 2012 8th International Conference on Natural Computation, pp. 79-83. IEEE, 2012.

Wu, Jui-Yu, and Chi-Jie Lu. "Computational intelligence approaches for stock price forecasting." In 2012 International Symposium on Computer, Consumer and Control, pp. 52-55. IEEE, 2012.

Kao, Ling-Jing, Chih-Chou Chiu, Chi-Jie Lu, and Chih-Hsiang Chang. "A hybrid approach by integrating wavelet-based feature extraction with MARS and SVR for stock index forecasting." Decision Support Systems 54, no. 3 (2013): 1228-1244.

Soni, Payal, Yogya Tewari, and Deepa Krishnan. "Machine Learning approaches in stock price prediction: A systematic review." In Journal of Physics: Conference Series, vol. 2161, no. 1, p. 012065. IOP Publishing, 2022.

Mintarya, Latrisha N., Jeta NM Halim, Callista Angie, Said Achmad, and Aditya Kurniawan. "Machine learning approaches in stock market prediction: a systematic literature review." Procedia Computer Science 216 (2023): 96-102.

Patel, Ramkrishna, Vikas Choudhary, Deepika Saxena, and Ashutosh Kumar Singh. "Review of stock prediction using machine learning techniques." In 2021 5th International Conference on Trends in Electronics and Informatics (ICOEI), pp. 840-846. IEEE, 2021.

Gandhmal, Dattatray P., and K. Kumar. “Systematic analysis and review of stock market prediction techniques.” Computer Science Review 34 (2019): 100190.

Ray, Susmita. “A quick review of machine learning algorithms.” In 2019 International conference on machine learning, big data, cloud and parallel computing (COMITCon), pp. 35-39. IEEE, 2019.

Fischer, Thomas, and Christopher Krauss. "Deep learning with long short-term memory networks for financial market predictions." European journal of operational research 270, no. 2 (2018): 654-669.

Leung, Carson Kai-Sang, Richard Kyle MacKinnon, and Yang Wang. "A machine learning approach for stock price prediction." In Proceedings of the 18th International Database

https://www.kaggle.com/datasets/suruchiarora/yahoo-finance-dataset-2018-2023