pytorch实现用LSTM做股票价格预测
python3.0以上, pytorch 1.3.1, torchvision 0.4.1, Pillow 7.1.2, pandas 1.0.3
data目录:上证指数的csv文件 model目录:模型保存文件 dataset.py : 数据加载及预处理类,数据标准化、划分训练集及测试集等 evaluate.py : 预测 LSTMModel.py : 定义LSTM模型 parsermy.py : 常用参数 train.py:模型训练
直接运行train.py开始模型训练
直接运行evaluate.py开始模型预测