Predicting the direction of stock market prices using random forest

• DOI: 10.48550/arxiv.1605.00003
• Publication Date: 2016-01-01
• Authors: Khaidem, Luckyson; Saha, Snehanshu; Dey, Sudeepa Roy
• Keywords: Computational Engineering, Finance, and Science (cs.CE); FOS: Computer and information sciences; Computer Science - Machine Learning; 0202 electrical engineering, electronic engineering, information engineering; 02 engineering and technology; Computer Science - Computational Engineering, Finance, and Science; Machine Learning (cs.LG)

Predicting trends in stock market prices has been an area of interest for researchers for many years due to its complex and dynamic nature. Intrinsic volatility in stock market across the globe makes the task of prediction challenging. Forecasting and diffusion modeling, although effective can't be the panacea to the diverse range of problems encountered in prediction, short-term or otherwise. Market risk, strongly correlated with forecasting errors, needs to be minimized to ensure minimal risk in investment. The authors propose to minimize forecasting error by treating the forecasting problem as a classification problem, a popular suite of algorithms in Machine learning. In this paper, we propose a novel way to minimize the risk of investment in stock market by predicting the returns of a stock using a class of powerful machine learning algorithms known as ensemble learning. Some of the technical indicators such as Relative Strength Index (RSI), stochastic oscillator etc are used as inputs to train our model. The learning model used is an ensemble of multiple decision trees. The algorithm is shown to outperform existing algo- rithms found in the literature. Out of Bag (OOB) error estimates have been found to be encouraging. Key Words: Random Forest Classifier, stock price forecasting, Exponential smoothing, feature extraction, OOB error and convergence.