開(kāi)發(fā)環(huán)境
如果只運(yùn)行web項(xiàng)目,則只需安裝如下包:
- python 3.6.x
- django >= 2.1.4 (或者使用conda安裝最新版)
- pandas >= 0.23.4 (或者使用conda安裝最新版)
-
numpy >= 1.15.2 (或者使用conda安裝最新版)
*apscheduler = 2.1.2 (請(qǐng)用pip install apscheduler==2.1.2 安裝,conda裝的版本不兼容)
如果需要訓(xùn)練模型或者使用模型來(lái)預(yù)測(cè)(注:需要保證本機(jī)擁有 NVIDIA GPU以及顯卡驅(qū)動(dòng)),則還需要安裝:
- tensorflow-gpu >= 1.10.0 (可以使用conda安裝最新版。如用conda安裝,cudatoolkit和cudnn會(huì)被自動(dòng)安裝)
- cudatoolkit >= 9.0 (根據(jù)自己本機(jī)的顯卡型號(hào)決定,請(qǐng)去NVIDIA官網(wǎng)查看)
- cudnn >= 7.1.4 (版本與cudatoolkit9.0對(duì)應(yīng)的,其他版本請(qǐng)去NVIDIA官網(wǎng)查看對(duì)應(yīng)的cudatoolkit版本)
- keras >= 2.2.2 (可以使用conda安裝最新版)
- matplotlib >= 2.2.2 (可以使用conda安裝最新版)
數(shù)據(jù)獲取
訓(xùn)練模型的數(shù)據(jù),即10個(gè)公司的歷史股票數(shù)據(jù)。獲取國(guó)內(nèi)上市公司歷史股票數(shù)據(jù), 并以csv格式保存下來(lái)。csv格式方便用pandas讀取,輸入到LSTM神經(jīng)網(wǎng)絡(luò)模型, 用于訓(xùn)練模型以及預(yù)測(cè)股票數(shù)據(jù)。