텐서플로 에러 KeyError: 'val_loss'

에러

KeyError: 'val_loss'

 

코드

>>># from keras.optimizers import RMSprop
>>>from tensorflow.keras.optimizers import RMSprop

>>>model = Sequential()
>>>model.add(layers.Flatten(input_shape=(lookback // step, float_data.shape[-1])))
>>>model.add(layers.Dense(32, activation='relu'))
>>>model.add(layers.Dense(1))

 

>>>model.compile(optimizer=RMSprop(), loss='mae')
>>>history = model.fit_generator(train_gen,
>>>                              steps_per_epoch=500,
>>>                              epochs=20,
>>>                              validation_data=val_gen,
>>>                              validation_steps=val_steps)
                              
>>>loss = history.history['loss']
>>>val_loss = history.history['val_loss']

 

원인

이 문제는 RMSprop 문제로 compile 이 제대로 작동하지 않아서 발생하는 문제임. 

 

해결 

아래 문장으로 대체

>>> from tensorflow.keras.optimizers import RMSprop