TensorFlow

TensorFlow - validation_data 파라미터

Cong_S 2022. 6. 14. 11:36

에포크 시마다 테스트를 하는, 밸리데이션 데이터를 처리하는 방법 중에 하나인 validation_data 에 대해 알아보자.

epoch_history = model.fit(X_train, y_train, epochs= 30 , validation_data=(X_test, y_test), callbacks=[ealry_stop])

매 에포크마다 기존 학습데이터가 아닌 새로운 데이터로 성능을 평가하는 

밸리데이션 데이터를 처리하는 파라미터 중

validation_data 의 특징은 사용자가 이미 데이터셋을 준비해놓아야 사용이 가능하다는 점이다. 

 

validation_split 같은 경우는 현재 학습 데이터 중 일부를 쪼개 밸리데이션의 평가 데이터로  사용하는 것과 달리

이미 학습에 사용되지 않은 새로운 데이터셋이 준비되어있고 그 X와 y 데이터를 파라미터에 입력해 사용하면 된다.