Machine learning and deep learning models for stock price prediction, case study: Google Company
- 1 Rensselaer Polytechnic Institute
* Author to whom correspondence should be addressed.
Abstract
Stock prediction is crucial for investors to manage risk, diversify portfolios, and plan for long-term financial goals. Companies use predictions to allocate capital wisely, attract investors, and make strategic decisions. Stock prediction is to analyze competitive stock (e.g. Google, Apple and Microsoft), involving parameters of revenue, profit, PE ratio, growth rate, and to predict the future price of the stock. Based on these assumptions, this project objective is to forecast the trend in Google stock prices with four predictive models: linear regression, XGBoost, Long Short-Term Memory Network (LSTM) and Recurrent neural network (RNN). The dataset is Google stock prediction dataset, which involves 17,598 stock information. Among all four models, LSTM method obtains the best final result with 0.0012 of MSE and 0.95115 of R square. LSTM has the lowest MSE, and its R square is similar to that of other models. For future plan, this project may expand more on feature engineering, model techniques and fine-tuning methods.
Keywords
Machine learning, deep learning, stock price forecast, neural network
[1]. R. Patel, V. Choudhary, D. Saxena and A. K. Singh, " Review of stock prediction using machine learning techniques," 2021 Conference on Trends in Electronics and Informatics, 2021, pp. 840-846.
[2]. Tang, H., Chiu, K. C., & Xu, L. (2003). Finite mixture of ARMA-GARCH model for stock price prediction. Computational Intelligence in Economics and Finance, pp. 1112-1119.
[3]. Y. E. Cakra and B. Distiawan Trisedya, "Stock price prediction using linear regression based on sentiment analysis," 2015 International Conference on Advanced Computer Science and Information Systems, 2015, pp. 147-154.
[4]. L. Jidong and Z. Ran, "Dynamic Weighting Multi Factor Stock Selection Strategy Based on XGboost Machine Learning Algorithm," 2018 Conference of Safety Produce Informatization, 2018, pp. 868-872.
[5]. Zhao, J., Zeng, D., Liang, S. et al. Prediction model for stock price trend based on recurrent neural network. J Ambient Intell Human Comput 12, 745–753 (2021)
[6]. Roondiwala, M., Patel, H., & Varma, S. (2017). Predicting stock prices using LSTM. International Journal of Science and Research, 6(4), 1754-1756.
[7]. NVIDIA., “What is XGBoost?” NVIDIA Data Science Glossary, (2022). https://www.nvidia.com/en-us/glossary/data-science/xgboost/
[8]. Chen, T., & Guestrin, C., "Xgboost: A scalable tree boosting system." Acm sigkdd international conference on knowledge discovery and data mining, 785-794 (2016).
[9]. Lipton, Zachary C., John Berkowitz, and Charles Elkan. "A critical review of recurrent neural networks for sequence learning." arXiv preprint arXiv:1506.00019 (2015).
[10]. Roondiwala, Murtaza and Harshal Patel "Predicting stock prices using LSTM." International Journal of Science and Research 6.4 (2017): 1754-1756.
[11]. S. Hochreiter and J. Schmidhuber, "Long Short-Term Memory," in Neural Computation, vol. 9, no. 8, pp. 1735-1780, 1997.
Cite this article
Wang,X. (2024). Machine learning and deep learning models for stock price prediction, case study: Google Company. Applied and Computational Engineering,49,254-264.
Data availability
The datasets used and/or analyzed during the current study will be available from the authors upon reasonable request.
Disclaimer/Publisher's Note
The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of EWA Publishing and/or the editor(s). EWA Publishing and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content.
About volume
Volume title: Proceedings of the 4th International Conference on Signal Processing and Machine Learning
© 2024 by the author(s). Licensee EWA Publishing, Oxford, UK. This article is an open access article distributed under the terms and
conditions of the Creative Commons Attribution (CC BY) license. Authors who
publish this series agree to the following terms:
1. Authors retain copyright and grant the series right of first publication with the work simultaneously licensed under a Creative Commons
Attribution License that allows others to share the work with an acknowledgment of the work's authorship and initial publication in this
series.
2. Authors are able to enter into separate, additional contractual arrangements for the non-exclusive distribution of the series's published
version of the work (e.g., post it to an institutional repository or publish it in a book), with an acknowledgment of its initial
publication in this series.
3. Authors are permitted and encouraged to post their work online (e.g., in institutional repositories or on their website) prior to and
during the submission process, as it can lead to productive exchanges, as well as earlier and greater citation of published work (See
Open access policy for details).