본문 바로가기

Deep Learning/Tensorflow

Tensorflow - validation_data 파라미터 사용 법

반응형
validation_data 파라미터 사용 법

 

 

 

에포크시마다 테스트를 하는, 밸리데이션 데이터를 처리하는 방법 중 하나로

 

validation_split은 train data의 몇퍼센트를 밸리데이션 데이터로 처리할 것인가를 나타내는데

 

vaildation_data는 data를 지정해 줌으로써 test data로 중간평가를 진행할 수 있다.

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

 

반응형