Improving stock trading decisions based on pattern recognition using machine learning technology

PLoS One. 2021 Aug 6;16(8):e0255558. doi: 10.1371/journal.pone.0255558. eCollection 2021.

Abstract

PRML, a novel candlestick pattern recognition model using machine learning methods, is proposed to improve stock trading decisions. Four popular machine learning methods and 11 different features types are applied to all possible combinations of daily patterns to start the pattern recognition schedule. Different time windows from one to ten days are used to detect the prediction effect at different periods. An investment strategy is constructed according to the identified candlestick patterns and suitable time window. We deploy PRML for the forecast of all Chinese market stocks from Jan 1, 2000 until Oct 30, 2020. Among them, the data from Jan 1, 2000 to Dec 31, 2014 is used as the training data set, and the data set from Jan 1, 2015 to Oct 30, 2020 is used to verify the forecasting effect. Empirical results show that the two-day candlestick patterns after filtering have the best prediction effect when forecasting one day ahead; these patterns obtain an average annual return, an annual Sharpe ratio, and an information ratio as high as 36.73%, 0.81, and 2.37, respectively. After screening, three-day candlestick patterns also present a beneficial effect when forecasting one day ahead in that these patterns show stable characteristics. Two other popular machine learning methods, multilayer perceptron network and long short-term memory neural networks, are applied to the pattern recognition framework to evaluate the dependency of the prediction model. A transaction cost of 0.2% is considered on the two-day patterns predicting one day ahead, thus confirming the profitability. Empirical results show that applying different machine learning methods to two-day and three-day patterns for one-day-ahead forecasts can be profitable.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Commerce / economics*
  • Commerce / statistics & numerical data
  • Decision Making*
  • Forecasting / methods*
  • Investments / economics*
  • Investments / statistics & numerical data
  • Machine Learning*
  • Models, Economic*
  • Neural Networks, Computer

Grants and funding

Haijun Yang: Grant No. 71771006 Shangcun Liu: Grant No. 71771008 The authors declare that they have no competing financial interests.This research was partially supported by National Natural Science Foundation of China (Grant No. 71771006 and 71771008). There was no additional external funding received for this study.