博主路一瓢的博文:
Tensorflow實例:利用LSTM預測股票每日最高價(一)
Tensorflow實例:利用LSTM預測股票每日最高價(二)
在以上兩篇博文中,博主介紹了RNN和LSTM,並使用LSTM實現了股票預測的例子,讓人受益匪淺。
但博主提供的代碼有一些bug,無法直接運行,經過本人的多番調試,已能完整運行,希望能在深度學習的路上給大家帶來一些幫助。
以上爲原博文
之前的代碼結構有些混亂,深度學習框架也有了長足的進步,遂重寫了相關代碼,使其具備如下特性:
- 更簡潔更模塊化
- 支持pytorch,keras和tensorflow三個主流深度學習框架
- 參數、模型、框架可高度定製化修改
- 支持增量式訓練
- 支持同時預測多個指標,預測未來任意天數
- 更多特性,詳解GitHub代碼:stock_predict_with_LSTM
注:tensorflow是原版的代碼,不具備上述特性,tensorflow真不好用,就不改了【狗頭保命】。
基於pytorch框架,同時預測未來一天的最高價和最低價的結果如下:
代碼在更新之後,很多之前評論區的問題可以得到快速的解決
如果仍有問題,可以在W XG ZH:海晨威,聯繫到本人,知無不言
G Z H,剛開啓,小萌新,
寫算法,記隨筆,不定期。
如分享,幫到你,實我幸,
感興趣,多支持,先謝過。
—— 海晨威