Skip to content

Latest commit

 

History

History
11 lines (11 loc) · 599 Bytes

README.md

File metadata and controls

11 lines (11 loc) · 599 Bytes

LSTM_LRP_factor

使用LSTM及股票因子数据预测未来收益,使用LRP(layer-wise relevance propagation)增强网络可解释性

模型结构

包括数据预处理、模型训练测试及LRP反向传播三个部分

代码框架

preprocessing.py:因子数据预处理文件;
main.py:主模型入口;
datasets.py:特征向量处理;
model.py:LSTM模型,包含模型训练、测试及LRP;
LRP_linear_layer.py:LRP计算公式
utils.py:基本函数;