我想知道是否可以打印RNN的指标,详细模式(verbose)只显示了一个简化的数字,例如“0.0014”,但我想打印完整的最终值。我希望能够打印出指标(均方误差,mse)的最终值。
这是我的模型代码:
# 使用Keras创建模型
model10 = Sequential()
model10.add(GRU(units=120, return_sequences=True, input_shape=(1,12),activity_regularizer=regularizers.l2(0.0005)))
model10.add(GRU(units=80, return_sequences=True,dropout=0.1))
model10.add(GRU(units=40, dropout=0.1))
model10.add(Dense(units=5))
model10.add(Dense(units=3))
model10.add(Dense(units=1, activation='relu'))
model10.compile(loss=['mae'], optimizer=Adam(lr=0.0005),metrics=['mse'])
model10.summary()
history10=model10.fit(X10_train, y10_train, batch_size=1000,epochs=20,validation_split=0.1, verbose=1, callbacks=[TensorBoardColabCallback(tbc),Early_Stop])
谢谢。
回答:
要获取最终的MSE值,可以尝试:
history10.history['mean_squared_error'][-1]
要获取整个历史记录,即每个epoch的值,可以尝试:
history10.history['mean_squared_error']