딥러닝

딥러닝(10) 에포크시마다 테스트를 하는, 벨리데이션 데이터를 처리하는 방법중 validation_data 파라미터 사용법

개발연습자1 2022. 12. 29. 18:06

 

제목은 길지만 사실 코드 하나만 바꾸면 된다. validation 파라미터로 validation_split을 쓰는 회사가 있고  validation_data를 파라 미터로 쓰는 회사가 있기 때문이다.

 

어떤 코드만 바꾸면 되냐면

# 학습시 validation_split 대신 validation_data를 써준다.
# validation_data의 파라미터는 X_test,y_test이다.

epoch_history = model.fit(X_train,y_train,epochs=10,
                         validation_data=(X_test,y_test))

validation_data는 파라미터로 X_test,y_test 쓴다.

반응형