Keras:시각화

Sam (토론 | 기여)님의 2021년 12월 6일 (월) 13:50 판 (새 문서: == 개요 == 각 에포크당 어떻게 학습이 이루어지는지 시각화 하면 모델을 이해하는 데 좋다. <syntaxhighlight lang="python"> import matplotlib.pyplot as p...)
(차이) ← 이전 판 | 최신판 (차이) | 다음 판 → (차이)

개요

각 에포크당 어떻게 학습이 이루어지는지 시각화 하면 모델을 이해하는 데 좋다.

import matplotlib.pyplot as plt

acc = history.history['accuracy']  # history.history에 학습정보가 담긴다.
val_acc = history.history['val_accuracy']
loss = history.history['loss']
val_loss = history.history['val_loss']

epochs = range(1, len(acc)+1)  # x축에 사용할 에폭을 정의한다.
plt.plot(epochs, loss, 'bo', label='Training loss')  # bo는 파란 점을 의미.
plt.plot(epochs, val_loss, 'b', label='Validation loss')  # b는 파란 실선을 의미.
plt.xlable('Epochs')
plt.ylable('Loss')
plt.show()