A Sales Prediction Method Based on LSTM with Hyper-Parameter Search. Issue 1 (February 2021)

• Record Type: Journal Article
• Authors: Dai, Yun; Huang, Jinghao
• Is Part Of: Journal of physics. Volume 1756:Issue 1(2021)
• Publication Date: 2021-02
• Subjects: Physics -- Congresses; 530.5
• DOI: 10.1088/1742-6596/1756/1/012015
• Languages: English
• ISSNs: 1742-6588

Abstract

Abstract: Sales forecast is a significant topic in business operation, which generally formulated as a time-series regression problem. Although there are many research results in this field, we still face some challenges in real scenes, such as data with high-sparsity, users may have a preference in prediction results, and systems need a single model with high performance. In this paper, a method is proposed to address the above challenges. We present a long short-time memory (LSTM) model with a special loss function and use the hyper-parameter search for accuracy optimization. To illustrate the performance, we employ them on the open dataset, Kaggle Rossman sales data. The experiment results show that compare with a series of machine learning models using the AutoML (Auto Machine Learning) tool, the proposed method significantly increased the performance of prediction on sparse data. Besides, it can reasonably overestimate or underestimate sales forecasts based on user preferences that meet the actual business demands.