问题描述
history = compile_and_fit(lstm_model,wide_window)
IPython.display.clear_output()
val_performance['LSTM'] = lstm_model.evaluate(wide_window.val)
performance['LSTM'] = lstm_model.evaluate(wide_window.test,verbose=0)
当我运行此代码时,它给我一个错误,即“未定义compile_and_fit”,而在tensorflow的colab文档中可执行相同的代码。
解决方法
MAX_EPOCHS = 20
def compile_and_fit(model,window,patience=2):
early_stopping = tf.keras.callbacks.EarlyStopping(monitor='val_loss',patience=patience,mode='min')
model.compile(loss=tf.losses.MeanSquaredError(),optimizer=tf.optimizers.Adam(),metrics=[tf.metrics.MeanAbsoluteError()])
history = model.fit(window.train,epochs=MAX_EPOCHS,validation_data=window.val,callbacks=[early_stopping])
return history
来源:https://www.tensorflow.org/tutorials/structured_data/time_series#linear_model