인프런 커뮤니티 질문&답변

hyunpung님의 프로필 이미지
hyunpung

작성한 질문수

[개정판] 파이썬 머신러닝 완벽 가이드

섹션2 Grid Search 예제에서 Test data분리시 계층 적용 여부

작성

·

55

0

섹션2의 Grid Search 예제에서 학습/테스트 데이터 분리시 train_test_split()함수에서 stratify옵션없이 사용되었는데요, 계층 분할을 위해 stratify=iris_data.target 옵션을 넣어야 하는거 아닌가요?

stratify옵션 넣고 테스트해보니 학습데이터 score는 강의동영상의 점수보다 낮았는데, test data에 대한 스코어는 강의 동영상과 동일하게 나왔습니다.

 

답변 1

0

권 철민님의 프로필 이미지
권 철민
지식공유자

안녕하십니까,

네, 좀더 정확하게 하려면 stratify 옵션을 주는게 좋습니다. 다만 현 실습 코드에는 크게 영향이 없어서 넣지 않았습니다.

감사합니다.

 

hyunpung님의 프로필 이미지
hyunpung

작성한 질문수

질문하기