목록overfitting 방지 (1)

코딩뚠뚠

[머신러닝 공부] K-Fold Cross Validation 이란

개요 : 딥러닝 모델을 Training 할 때 Train Data를 Train set 과 Validation set 으로 나눠줄 필요가 있다. Keras를 이용할 때에는 validation_split 으로 나눠줄 수 있었다. keras.preprocessing.image.ImageDataGenerator (validation_split=0.7, .... etc ....) 하지만 augmentation을 하더라도 기본 데이터가 다소 적을 때 validation_split은 Train Data 전부를 사용하지 못했다. sklearn의 train_test_split() 은 random으로 validation set을 선택하지만 이또한 validation에 사용하는 데이터를 train에 사용할 수는 없다는 단점..
공부/ML&DL 2021. 4. 18. 01:23
이전 Prev 1 Next 다음