test_size = 0.3으로 지정했기 때문에 와인 샘플의 30%가 X_test와 y_test에 할당됩니다. 나머지 샘플 70%는 X_train과 y_train에 각각 할당됩니다. stratify 매개변수에 클래스 레이블 배열 y를 전달하면 훈련 데이터셋과 테스트 데이터셋에 있는 클래스 비율이 원본 데이터셋과 동일하게 유지됩니다.
추천 도서와 신규 콘텐츠를 먼저 받아보세요
엄선한 추천 도서와 신규 콘텐츠를 이메일로 보내드립니다. 새로운 책이 등록될 때마다 소식을 받아보세요.