그다음 모든 폴드에 대해 에포크의 MAE 점수 평균을 계산합니다.
코드 4-28 K-겹 검증 점수 평균 기록하기
average_mae_history = [
np.mean([x[i] for x in all_mae_histories]) for i in range(num_epochs)]
그래프로 나타내면 그림 4-9와 같습니다.
코드 4-29 검증 점수 그래프 그리기
plt.plot(range(1, len(average_mae_history) + 1), average_mae_history)
plt.xlabel("Epochs")
plt.ylabel("Validation MAE")
plt.show()
▲ 그림 4-9 에포크별 검증 MAE