overfiting
-
[DeepLearning] 학습 단계를 train, validation, test 로 나누는 이유AI 2020. 4. 20. 21:33
pytorch로 예제를 학습하는 도중에 익숙치 않아서 그런지 사소한 개념이 적립에 시간이 많이 소요되고 있다. 그 중 하나가 모델을 학습할때 train, validation, test로 단계를 나누는 이유이다. pytorch에선 아래 코드로 모드를 변경 시킬 수 있고, 꼭 해야한다. model.train() # 트레이닝모드 model.eval() # 검증모드 model.eval()의 경우 validation과 test 단계에 꼭 선언해야한다. 그럼 개념적으로 정리를 해보자. 데이터가 총 1000개가 있다면, train 6000개, validation 2000개, test 2000개로 나누어 준다. train 단계는 말 그대로 모델을 학습시키는 단계이다. 그 말은 즉 validation과 test 단계에서..