Tensorflow 훈련 과정 시각화

tensorflow에서 훈련 과정을 시각화하기 위해 아래와 같은 코드를 작성 하였다

 

import matplotlib.pyplot as plt

~~~~~~
~~~~~~

hist = train_model.fit(~~~~~)

fig, loss_ax = plt.subplots()
acc_ax = loss_ax.twinx()

loss_ax.plot(hist.history['loss'], 'y', label='train loss')

loss_ax.plot(hist.history['val_loss'], 'r', label='val loss')

acc_ax.plot(hist.history['psnr'], 'b', label='train psnr')

acc_ax.plot(hist.history['val_psnr'], 'g', label='val psnr')

loss_ax.set_xlabel('epoch')

loss_ax.set_ylabel('loss')

acc_ax.set_ylabel('psnr')

loss_ax.legend(loc='upper left')

acc_ax.legend(loc='lower left')

plt.show()