Cargando…
Stock prediction based on bidirectional gated recurrent unit with convolutional neural network and feature selection
With the development of recent years, the field of deep learning has made great progress. Compared with the traditional machine learning algorithm, deep learning can better find the rules in the data and achieve better fitting effect. In this paper, we propose a hybrid stock forecasting model based...
Autores principales: | , , |
---|---|
Formato: | Online Artículo Texto |
Lenguaje: | English |
Publicado: |
Public Library of Science
2022
|
Materias: | |
Acceso en línea: | https://www.ncbi.nlm.nih.gov/pmc/articles/PMC8815979/ https://www.ncbi.nlm.nih.gov/pubmed/35120138 http://dx.doi.org/10.1371/journal.pone.0262501 |
_version_ | 1784645349829246976 |
---|---|
author | Zhou, Qihang Zhou, Changjun Wang, Xiao |
author_facet | Zhou, Qihang Zhou, Changjun Wang, Xiao |
author_sort | Zhou, Qihang |
collection | PubMed |
description | With the development of recent years, the field of deep learning has made great progress. Compared with the traditional machine learning algorithm, deep learning can better find the rules in the data and achieve better fitting effect. In this paper, we propose a hybrid stock forecasting model based on Feature Selection, Convolutional Neural Network and Bidirectional Gated Recurrent Unit (FS-CNN-BGRU). Feature Selection (FS) can select the data with better performance for the results as the input data after data normalization. Convolutional Neural Network (CNN) is responsible for feature extraction. It can extract the local features of the data, pay attention to more local information, and reduce the amount of calculation. The Bidirectional Gated Recurrent Unit (BGRU) can process the data with time series, so that it can have better performance for the data with time series attributes. In the experiment, we used single CNN, LSTM and GRU models and mixed models CNN-LSTM, CNN-GRU and FS-CNN-BGRU (the model used in this manuscript). The results show that the performance of the hybrid model (FS-CNN-BGRU) is better than other single models, which has a certain reference value. |
format | Online Article Text |
id | pubmed-8815979 |
institution | National Center for Biotechnology Information |
language | English |
publishDate | 2022 |
publisher | Public Library of Science |
record_format | MEDLINE/PubMed |
spelling | pubmed-88159792022-02-05 Stock prediction based on bidirectional gated recurrent unit with convolutional neural network and feature selection Zhou, Qihang Zhou, Changjun Wang, Xiao PLoS One Research Article With the development of recent years, the field of deep learning has made great progress. Compared with the traditional machine learning algorithm, deep learning can better find the rules in the data and achieve better fitting effect. In this paper, we propose a hybrid stock forecasting model based on Feature Selection, Convolutional Neural Network and Bidirectional Gated Recurrent Unit (FS-CNN-BGRU). Feature Selection (FS) can select the data with better performance for the results as the input data after data normalization. Convolutional Neural Network (CNN) is responsible for feature extraction. It can extract the local features of the data, pay attention to more local information, and reduce the amount of calculation. The Bidirectional Gated Recurrent Unit (BGRU) can process the data with time series, so that it can have better performance for the data with time series attributes. In the experiment, we used single CNN, LSTM and GRU models and mixed models CNN-LSTM, CNN-GRU and FS-CNN-BGRU (the model used in this manuscript). The results show that the performance of the hybrid model (FS-CNN-BGRU) is better than other single models, which has a certain reference value. Public Library of Science 2022-02-04 /pmc/articles/PMC8815979/ /pubmed/35120138 http://dx.doi.org/10.1371/journal.pone.0262501 Text en © 2022 Zhou et al https://creativecommons.org/licenses/by/4.0/This is an open access article distributed under the terms of the Creative Commons Attribution License (https://creativecommons.org/licenses/by/4.0/) , which permits unrestricted use, distribution, and reproduction in any medium, provided the original author and source are credited. |
spellingShingle | Research Article Zhou, Qihang Zhou, Changjun Wang, Xiao Stock prediction based on bidirectional gated recurrent unit with convolutional neural network and feature selection |
title | Stock prediction based on bidirectional gated recurrent unit with convolutional neural network and feature selection |
title_full | Stock prediction based on bidirectional gated recurrent unit with convolutional neural network and feature selection |
title_fullStr | Stock prediction based on bidirectional gated recurrent unit with convolutional neural network and feature selection |
title_full_unstemmed | Stock prediction based on bidirectional gated recurrent unit with convolutional neural network and feature selection |
title_short | Stock prediction based on bidirectional gated recurrent unit with convolutional neural network and feature selection |
title_sort | stock prediction based on bidirectional gated recurrent unit with convolutional neural network and feature selection |
topic | Research Article |
url | https://www.ncbi.nlm.nih.gov/pmc/articles/PMC8815979/ https://www.ncbi.nlm.nih.gov/pubmed/35120138 http://dx.doi.org/10.1371/journal.pone.0262501 |
work_keys_str_mv | AT zhouqihang stockpredictionbasedonbidirectionalgatedrecurrentunitwithconvolutionalneuralnetworkandfeatureselection AT zhouchangjun stockpredictionbasedonbidirectionalgatedrecurrentunitwithconvolutionalneuralnetworkandfeatureselection AT wangxiao stockpredictionbasedonbidirectionalgatedrecurrentunitwithconvolutionalneuralnetworkandfeatureselection |