使用pytorch框架lstm模型预测股价,通过tushare获取平安银行5000个交易日收盘价数据,使用pytorch框架构建LSTM模型,隐藏层大小200,层数50,损失函数采用均方误差损失MSELOSS,优化器采用Adam,计算结果为后100个数值平均均方误差1.3。
评论