Integrating Clustering and Regularization for Robust LSTM-Based Stock Price Prediction

Stock price forecasting has oftentimes interested several researchers around the world. Making predictions for the future largely depends on the data that will be used to train the model. In general, historical data are used to train models, which contain a features of different types, out of which, not all are necessarily helpful in making predictions. It is, hence, crucial to select the features that can be most useful to make precise predictions. This article proposes a feature selection approach based on the K-means clustering algorithm and elastic net regularization. We have used the K-means algorithm to cluster all the correlated features together and apply elastic net regularization to select the most predictive features within each cluster. We use the selected features to train an LSTM model which predicts the future closing price of a stock for the upcoming trading day. We evaluate the performance of our proposed approach in comparison to the existing approach and observe performance improvement.