Vision AI/미분류

K-fold cross validation

beelinekim 2019. 11. 9. 22:23
반응형

교차 검증 기법 중 하나로, K개의 데이터 셋을 만들어서 K번만큼 각 셋에 학습과 검증평가를 반복적으로 수행하는 방법.

 

예를 들어, 150장의 영상 데이터로 5 fold cross validation을 한다.

 

그럼 training data는 120장, validation data는 30장이 된다.

 

5-fold는 전체 데이터 수를 5로 나눠서 training : validation = 4 : 1 ,

 

10-fold는 전체 데이터 수를 10으로 나눈 후 training : validation = 9 : 1

 

이런 식으로 나뉘어진다.

 

나눈 후, K번 만큼 training data, validation data를 바꿔가며 검증을 한다.

 

 

이런식으로 5번 검증을 수행하고, 5개의 예측 결과를 평균하여 k-fold validation 결과로 반영한다.

반응형