TechRxiv
Sales Forecasting using XGBoost Research Paper.docx.pdf (994.46 kB)

Sales Forecasting using XGBoost

Download (994.46 kB)
preprint
posted on 2022-11-07, 16:18 authored by Syed Hamad ul Haq AndrabiSyed Hamad ul Haq Andrabi, Dr K. Alice, Siddharth Anoop Srivastava

This study intends to investigate several machine learning algorithms for sales forecasting strategies. A retailer can use this to predict future market demand and adjust its inventory levels accordingly. The accuracy of these predictions will determine whether the retailer profits or suffers losses. In this paper, we worked on the Walmart Sales dataset from Kaggle. It has over 400,000 rows and about 20 columns. After cleaning and performing the necessary feature engineering of the data, we used machine learning algorithms such as eXtreme Gradient Boosting (with and without tuned hyperparameters), Linear Regression, Ridge Regression, Decision Tree Regressor and Random Forest Regressor (with and without tuned hyperparameters). The most effective algorithm out of all the others was XGBoost when the hyperparameters were tuned. This model performs well on sales prediction by utilising less processing power and memory. 

History

Email Address of Submitting Author

sa3184@srmist.edu.in

Submitting Author's Institution

SRM Institute of Science and Technology

Submitting Author's Country

  • India