| 注册
首页|期刊导航|计算机工程与应用|融合图卷积和卷积自注意力的股票预测方法

融合图卷积和卷积自注意力的股票预测方法

田红丽 崔姚 闫会强

计算机工程与应用2024,Vol.60Issue(4):192-199,8.
计算机工程与应用2024,Vol.60Issue(4):192-199,8.DOI:10.3778/j.issn.1002-8331.2210-0050

融合图卷积和卷积自注意力的股票预测方法

Stock Prediction Method Combining Graph Convolution and Convolution Self-Attention

田红丽 1崔姚 1闫会强2

作者信息

  • 1. 河北工业大学 人工智能与数据科学学院,天津 300401
  • 2. 河北工业大学 经济管理学院,天津 300401
  • 折叠

摘要

Abstract

With the continuous development of China's stock market,the trend of a stock is often affected by the devel-opment of the upstream and downstream industries of its enterprises.In view of the fact that the mainstream stock pre-diction model ignores the shortcomings of the correlation relationship between stocks,a stock trend prediction model fusing graph convolution and long convolution self-attention is proposed.Firstly,the relationship matrix of multiple associated stocks is calculated using the correlation coefficient,then the graph convolutional network combining rela-tionship matrix is used to extract the feature of the associated stocks.Secondly,the multi-head convolution is used to extract long-term features from attention.Finally,the classification loss function polynomial expansion framework is used to make trend prediction for loss function optimization.Experimental results show that the proposed model is superior to gated loop unit,time convolutional network and other models in terms of accuracy,precision,recall and F1 score.

关键词

股票趋势预测/卷积自注意力/去趋势互相关系数

Key words

stock trend forecasting/convolution self-attention/detrend the number of interrelationships

分类

信息技术与安全科学

引用本文复制引用

田红丽,崔姚,闫会强..融合图卷积和卷积自注意力的股票预测方法[J].计算机工程与应用,2024,60(4):192-199,8.

基金项目

国家社会科学基金(19BGL054). (19BGL054)

计算机工程与应用

OA北大核心CSTPCD

1002-8331

访问量0
|
下载量0
段落导航相关论文