Forecasting Stock Market Performance: An Ensemble Learning-Based Approach

Venkat Ramaraju, Jayanth Rao, James Smith, Ajay Bansal

Research output: Chapter in Book/Report/Conference proceedingConference contribution

Abstract

In a previous work, we presented Agora, a stock recommendation system based on sentiment analysis. One of the potential areas for improvement we recognized was the accuracy of the supervised machine learning model. We had previously employed a logistical regression machine learning model to generate our predictions. This paper aims to detail the improvements in accuracy we made by training and deploying ensemble models for our application. We detail our improved methodology in training Random Forest and XGBoost Classifier models on similar datasets from our original publication. We performed a comparison of the accuracies between the two models to show how our improved models lead to better results in stock market prediction. We have also provided sufficient context along the way to help a reader understand what we are attempting to achieve with this paper. Our Random Forest Classifier model outperformed the logistic regression model by 10.4%, which marked a significant improvement. Detailing how we beat our original accuracy is the major takeaway of this paper.

Original languageEnglish (US)
Title of host publicationProceedings - 2023 IEEE 6th International Conference on Artificial Intelligence and Knowledge Engineering, AIKE 2023
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages17-23
Number of pages7
ISBN (Electronic)9798350331288
DOIs
StatePublished - 2023
Event6th IEEE International Conference on Artificial Intelligence and Knowledge Engineering, AIKE 2023 -
Duration: Jan 1 2023 → …

Publication series

NameProceedings - 2023 IEEE 6th International Conference on Artificial Intelligence and Knowledge Engineering, AIKE 2023

Conference

Conference6th IEEE International Conference on Artificial Intelligence and Knowledge Engineering, AIKE 2023
Period1/1/23 → …

Keywords

  • Extreme Gradient Boosting
  • Random Forest
  • Sentiment Analysis
  • Stock Market
  • VADER

ASJC Scopus subject areas

  • Artificial Intelligence
  • Computer Vision and Pattern Recognition
  • Information Systems
  • Information Systems and Management
  • Statistics, Probability and Uncertainty
  • Health Informatics

Fingerprint

Dive into the research topics of 'Forecasting Stock Market Performance: An Ensemble Learning-Based Approach'. Together they form a unique fingerprint.

Cite this