개발/머신러닝

파이썬 머신러닝 - epochs history 로 차트 만들어 확인

웅'jk 2022. 12. 28. 13:12

만들어진 모델을 fit을 통해 학습한 결과를 변수에 저장하고 .history 를 통해 

출력되는 loss, metrics 값을 알 수 있습니다. 

 

이걸 차트로 표현합니다.

 

# 학습한 모델을 변수에 저장
hist = model.fit(X_train,y_train,batch_size = 10 , epochs= 20)

# 출력하면 loss,metrics 등의 딕셔너리로 출력
hist.history

# 차트로 표시하기 
import matplotlib.pyplot as plt
plt.plot(hist.history['loss'])
plt.xlabel('# epochs')
plt.ylabel('# loss')
plt.show