LSTM用于股票价格预测笔记,侵删。
原地址:https://blog.csdn.net/mylove0414/article/details/55805974
GitHub地址:https://github.com/LouisScorpio/datamining/tree/master/tensorflow-program/rnn/stock_predict
1. np.newaxis
np.newaxis的作用就是在其所在位置增加一个一维,如:
x1 = np.array([1, 2, 3, 4, 5])
# the shape of x1 is (5,)
x1_new = x1[:, np.newaxis]
# now, the shape of x1_new is (5, 1)
# array([[1],
# [2],
# [3],
# [4],
# [5]])
x1_new = x1[np.newaxis,:]
# now, the shape of x1_new is (1, 5)
# array([[1, 2, 3, 4, 5]])
2.array中的np.reshape(a,[-1,10])
数组中新的shape应该 要与原来的配套,如原来是一个10*10的数组,则转成三维,可以是2*2*25。
# a是一个10*10的array
np.reshape(a,[-1,2,25])
会自动转成2*2*25的数组,根据其他两个值计算-1位置的值。
3.TensorFlow中的saver
saver的作用是将训练好的模型