StockPrediction--自建的股票预测的python库

StockPrediction–自建的股票预测的python库

1. 安装

pip3 install StockPrediction

2. 使用方法

单个股票的历史收盘价数据，如平安银行

import StockPrediction
PingAn = StockPrediction.CNStocks('000001.sz')
PingAn.History()

有两个主要的类：CNStocks和USStocks，分别是中国股票和美股
中国股票分为上证和深证，注意股票代码后面的后缀
美股是四个字母，如苹果AAPL

Apple = StockPrediction.USStocks('AAPL')
Apple.History()

结果如下：

未来一周的收盘价的预测，如平安银行

PingAn.TrainAndPredict()

训练结果如下：

预测结果如下：

3. Tips

1. 获取股票数据使用的是yahoo的免费接口，不需要注册，apikey, token等，有一点就是没有VPN的话，尽量不要频繁访问，会提示timeout，遇到这种情况，过两三分钟访问就可以了。简单使用方法如下：

import pandas_datareader as web
df = web.DataReader('000001.sz', data_source='yahoo', start='2018-01-01', end='2021-08-12')
print(df)

2. 获取的数据包括开盘价，收盘价，当天最高等等，我是只提取了收盘价作为训练的数据，预测的也是收盘价
3. 历史数据的时间我是设置的从2018-01-01开始然后到当天，获取当天日期方法如下：

import datetime
date = datetime.date.today()

昨天和明天分别如下：

yesterday = today - datetime.timedelta(days=1)
tomorrow = today + datetime.timedelta(days=1)

4. 使用了RNN中的LSTM模型（两个LSTM层，两个Dense层），预测时采用了过去60天的收盘价
5. 只选用了收盘价，预测结果肯定有偏差，仅供参考和玩乐