分析这段代码 input_data = [train_x[:,:,i].reshape((train_x.shape[0],n_timesteps,1)) for i in range(n_features)]
时间: 2024-01-16 15:05:44 浏览: 134
Mnist_keras.py.zip_keras_kerasmnist_py神经网络_神经网络_神经网络 keras
这段代码是将三维的训练集数据(train_x)按特征逐个取出来,然后对每个特征进行reshape,变成一个三维的矩阵(shape为(train_x.shape[0],n_timesteps,1)),其中train_x.shape[0]表示样本数,n_timesteps表示时间步长,1表示特征维度。最终将所有特征的三维矩阵放入一个列表中,即input_data。
阅读全文