Does Machine Learning Algorithms Improve Forecasting Accuracy? Predicting Stock Market Index Using Ensemble Model

  • T. ViswanathanEmail author
  • Manu Stephen
Conference paper
Part of the Lecture Notes in Networks and Systems book series (LNNS, volume 127)


Forecasting market performance and understanding the mechanism of price discovery is inherent to develop trading strategies. This paper examines the predictive power of machine learning algorithms in forecasting stock index. The study applies various machine learning algorithms and suggests the best model for forecasting stock index. We have developed an ensemble model to predict the daily closing index of Nifty 50 based on open, high, low and previous day’s close. The ensemble model includes a mix of simple linear regression, gradient boosted tree, decision tree and random forest. The parameters of the models are tested for its accuracy train-test split under supervised learning. The predicting accuracy of machine learning algorithms is further refined using cross-validation techniques that include leaving one out cross-validation and k-fold cross-validation. We found that the ensemble model provides an accurate forecast of the stock market index for the short term. The outcome of the study would facilitate the investors and portfolio managers to use the appropriate model for forecasting and take an informed decision by considering the nature of stock market volatility.


Forecasting Linear regression Gradient boosted tree Decision tree and random forest Ensemble 


  1. 1.
    Garman MB, Klass MJ (1980) On the estimation of security price volatilities from historical data. J Bus 53(1):67–78Google Scholar
  2. 2.
    Alizadeh S, Brandt MW, Diebold FX (2002) Range-based estimation of stochastic volatility models. J Finan 57(3):1047–1091Google Scholar
  3. 3.
    Adebiyi AA, Adewumi AO, Ayo CK (2014) Comparison of ARIMA and artificial neural networks models for stock price prediction. J Appl Math 2014Google Scholar
  4. 4.
    Fischer T, Krauss C (2018) Deep learning with long short-term memory networks for financial market predictions. Eur J Oper Res 270(2):654–669MathSciNetCrossRefGoogle Scholar
  5. 5.
    Göçken M, Özçalıcı M, Boru A, Dosdoğru AT (2016) Integrating metaheuristics and artificial neural networks for improved stock price prediction. Expert Syst Appl 44:320–331CrossRefGoogle Scholar
  6. 6.
    Feuerriegel S, Gordon J (2018) Long-term stock index forecasting based on text mining of regulatory disclosures. Decis Support Syst 112:88–97CrossRefGoogle Scholar

Copyright information

© The Editor(s) (if applicable) and The Author(s), under exclusive license to Springer Nature Singapore Pte Ltd. 2021

Authors and Affiliations

  1. 1.Symbiosis Institute of Business ManagementBengaluruIndia

Personalised recommendations