반응형
데이터셋을 일정비율로 분리하는 : train_test_split 예제
Deep Learning & Machine Learning/강좌&예제 코드2023. 10. 17. 22:41데이터셋을 일정비율로 분리하는 : train_test_split 예제

train_test_split를 사용하여 데이터셋을 Train 데이터셋과 Test 데이터셋으로 분리하는 예제 코드입니다. 2023. 6. 11 최초작성 아래 링크에 있는 데이터셋의 Train 데이터셋을 분리하여 Train 데이터셋과 Test 데이터셋으로 분리했습니다. https://www.kaggle.com/datasets/andrewmvd/animal-faces 다음 처럼 train 데이터셋과 validation 데이터셋에 cat,dog,wild 클래스별 하위 디렉토리가 위치하고 각각의 파일에 이미지 파일이 포함되어 있습니다. 코드를 실행하면 train 디렉토리에 있는 클래스별 이미지 파일을 지정한 비율로 나누어서 dataset 하위 디렉토리에 train, test 디렉토리에 나누어서 저장합니다. 전체..

반응형
image