작성한 질문수
[개정판] 파이썬 머신러닝 완벽 가이드
작성
·
55
0
섹션2의 Grid Search 예제에서 학습/테스트 데이터 분리시 train_test_split()함수에서 stratify옵션없이 사용되었는데요, 계층 분할을 위해 stratify=iris_data.target 옵션을 넣어야 하는거 아닌가요?
stratify옵션 넣고 테스트해보니 학습데이터 score는 강의동영상의 점수보다 낮았는데, test data에 대한 스코어는 강의 동영상과 동일하게 나왔습니다.
답변 1
안녕하십니까,
네, 좀더 정확하게 하려면 stratify 옵션을 주는게 좋습니다. 다만 현 실습 코드에는 크게 영향이 없어서 넣지 않았습니다.
감사합니다.