在 keras 2.2.4 中进行时间序列预测训练时,我正在超越 epoch vs loss plot。模型配置 1 个 lstm 层,1 个密集层,num epochs - 64。在某些配置上,我得到了正确的图,只有两条曲线,一条用于验证集,一条用于训练损失数据集,而在某些配置上,我得到了这个荒谬的图在图像中。我无法理解为什么会这样?我的代码——
def train(trainingData, config):
inputShape, numNode, numEpoch, batchSize = config
if nDiff > 0:
trainingData = np.array(difference(trainingData))
trainX, trainY = trainingData[:, :-1], trainingData[:, -1]
trainX = trainX.reshape((trainX.shape[0], trainX.shape[1], 1))
model = Sequential()
model.add(LSTM(numNode, activation = 'relu', input_shape = (inputShape, 1)))
#model.add(Dense(4, activation = 'relu'))
model.add(Dense(1))
model.compile(loss = 'mse' , optimizer = 'adam')
history = model.fit(trainX, trainY, validation_split = 0.2, epochs = numEpoch, batch_size = batchSize, verbose = 0, shuffle = True)
# summarize history for loss
plt.plot(history.history['loss'])
plt.plot(history.history['val_loss'])
plt.title('model loss')
plt.ylabel('loss')
plt.xlabel('epoch')
plt.legend(['train', 'test'], loc='upper left')
plt.savefig(modelName + "ind")
return model