Stratified K-Fold

Stratified K-Fold 

특정 클래스로 분류하는 모델에 대해 K-Fold를 적용해서 Train data와 Validation data를 나눌 때 사용하는 방법이다

 

기존 K-Fold 방법을 그대로 가져가는데 하나 다른 점은 

 

데이터 마다 해당되는 클래스가 다를텐데

 

모든 Fold 안에 데이터가 속한 클래스 비율을 같게 하는 것이다

 

각 Validation 과정이 끝나면 성능에 대해 평가한 값을 평균을 내는 것 또한 K-Fold와 동일하다

 

이를 그림으로 나타내면 이렇다