만들어진 모델을 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
'개발 > 머신러닝' 카테고리의 다른 글
파이썬 머신러닝 - 이미지 학습을 위한 flatten, 3개 이상 분류 액티베이션함수 softmax 와 loss 셋팅방법. (0) | 2022.12.28 |
---|---|
파이썬 머신러닝 - learning rate 와 validation , earlystopping (1) | 2022.12.28 |
파이썬 머신러닝 - 텐서플로우를 이용한 regression 문제 모델링 (0) | 2022.12.28 |
파이썬 머신러닝 - 텐서플로우를 이용한 그리드서치 (0) | 2022.12.27 |
파이썬 머신러닝 - 텐서플로우를 이용한 ANN , Dummy variable trap,batch_size , epochs (0) | 2022.12.27 |