利用LSTM对股价进行预测并可视化

利用LSTM对股价进行预测并可视化

 

Tushare API

网址：Tushare大数据社区

注册账号并获取自己的token码

tushare提供了众多接口，根据积分不同，token权限不同，读者请自行前往官网了解，这里不做过多介绍。

首先导入要用到的包

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import tushare as ts
from keras.layers import LSTM, Dense, Dropout
from keras.models import Sequential, load_model
from sklearn.preprocessing import MinMaxScaler
import talib
import mpl_finance as mpf

 调用tushare API获取股价数据

token = '你的token码'
ts.set_token(token)
data_ts = ts.pro_bar(ts_code='000001.SZ', start_date='20100101', end_date='20201214', asset='E', freq='D').iloc[::-1, :]
data_ts = data_ts.reset_index()
data_ts = data_ts.drop('index', axis=1)

对2020年的数据进行可视化

数据处理

## K线
data_plot = data_ts.loc[data_ts['trade_date'] >= '20200101']
data_plot = data_plot.reset_index()
data_plot = data_plot.drop(['index'], ax