stockPredict

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：模型训练

LSTM:

直接运行train.py开始模型训练

直接运行evaluate.py开始模型预测

RNN_BASE:

直接运行train_RNN.py开始模型训练

直接运行evaluate_RNN.py开始模型预测

MLP:

运行 MLP.py

DA_RNN:

直接运行train_DARNN.py开始模型训练

直接运行evaluate_DARNN.py开始模型预测

Name		Name	Last commit message	Last commit date
Latest commit History 6 Commits
__pycache__		__pycache__
data		data
img		img
model		model
.gitignore		.gitignore
DA_RNN.py		DA_RNN.py
LSTMModel.py		LSTMModel.py
LSTMModel_org.py		LSTMModel_org.py
MLP.py		MLP.py
README.md		README.md
RNN_base.py		RNN_base.py
all.log		all.log
all.log.2022-05-17		all.log.2022-05-17
dataset.py		dataset.py
evaluate.py		evaluate.py
evaluate_DARNN.py		evaluate_DARNN.py
evaluate_RNN.py		evaluate_RNN.py
log.py		log.py
parser_my.py		parser_my.py
self_test.py		self_test.py
train.py		train.py
train_DARNN.py		train_DARNN.py
train_RNN.py		train_RNN.py
未命名.ipynb		未命名.ipynb