강의

멘토링

커뮤니티

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

김완수님의 프로필 이미지
김완수

작성한 질문수

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

GridSearchCV 관련 내용문의

작성

·

253

0

안녕하세요. 

개정판 책의 내용 중 GridSearchCV에 대해 공부를 하다가 궁금한 점이 생겨 질문 남깁니다.

cross_val_score() API는 cross_validation을 할때 Stratified KFold를 사용한다고 말씀해주셨는데,

GridSearchCV의 파라미터인 cv에 KFold나 StratifiedKFold를 사용하지 않고 단순 폴딩 개수만 지정한다면

어떤 방식으로 cross_validation을 수행하는지 궁금합니다.

또 하나 질문으로는 cv=KFold or StratifidedKFold를 사용해도 괜찮은지와 유의사항이 있는지도 궁금합니다.

항상 좋은 강의 및 자료에 감사합니다!

답변 1

0

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

안녕하십니까, 

기본적으로 GridSearchCV의 cv은 StratifiedKFold를 적용합니다. cv=3 과 같이 설정하면 StratifiedKFold를 기반으로 합니다. 

그런데 KFold를 할수는 있습니다(굳이 할 이유는 적지만). 적어주신대로 

cv=KFold(n_splits=3).split(train) 하시면 됩니다. 

김완수님의 프로필 이미지
김완수

작성한 질문수

질문하기