开发环境
如果只运行web项目,则只需安装如下包:
- python 3.6.x
- django >= 2.1.4 (或者使用conda安装最新版)
- pandas >= 0.23.4 (或者使用conda安装最新版)
-
numpy >= 1.15.2 (或者使用conda安装最新版)
*apscheduler = 2.1.2 (请用pip install apscheduler==2.1.2 安装,conda装的版本不兼容)
如果需要训练模型或者使用模型来预测(注:需要保证本机拥有 NVIDIA GPU以及显卡驱动),则还需要安装:
- tensorflow-gpu >= 1.10.0 (可以使用conda安装最新版。如用conda安装,cudatoolkit和cudnn会被自动安装)
- cudatoolkit >= 9.0 (根据自己本机的显卡型号决定,请去NVIDIA官网查看)
- cudnn >= 7.1.4 (版本与cudatoolkit9.0对应的,其他版本请去NVIDIA官网查看对应的cudatoolkit版本)
- keras >= 2.2.2 (可以使用conda安装最新版)
- matplotlib >= 2.2.2 (可以使用conda安装最新版)
数据获取
训练模型的数据,即10个公司的历史股票数据。获取国内上市公司历史股票数据, 并以csv格式保存下来。csv格式方便用pandas读取,输入到LSTM神经网络模型, 用于训练模型以及预测股票数据。