선생님 교차검증 질문입니다.

12 views
Skip to first unread message

Chang-Hyeon Hwang (mayaoriginal)

unread,
Jan 27, 2024, 12:44:13 AMJan 27
to 머신러닝/딥러닝 도서 Q&A
교차검증을 수행할때, 궁금함이 있어 질의 드립니다.

<1>
from sklearn.model_selection import cross_validate
score=cross_validate(dt,train_input,train_target)
score
print(np.mean(score['test_score']))


<2>
from sklearn.model_selection import StratifiedKFold
score=cross_validate(dt,train_input,train_target,cv=StratifiedKFold())
score
print(np.mean(score['test_score']))

두개의 코딩 중, 기본적으로 분류일때는 StratifiedKFold를 사용한다고 하셨는데, <2>처럼 코딩안해도 기본적으로 StratifiedKFold를 사용하므로 <1>처럼 코딩해도 되는 건가요? 

혹, splitter를 따로 선언하여 10-폴드를 설명하시기위해 예로 드신건지요?

Haesun Park

unread,
Jan 28, 2024, 2:22:28 AMJan 28
to Chang-Hyeon Hwang (mayaoriginal), 머신러닝/딥러닝 도서 Q&A
안녕하세요. 박해선입니다.
cross_val_score 함수는 추정기가 분류 모델일 때 자동으로 StratifiedKFold를 사용하므로 스플리터 객체를 따로 만들어 넣어 줄 필요가 없습니다.
만약 기본 5폴드가 아니라 10폴드 교차 검증을 수행하려면 스플리터를 만들어 사용하세요.
감사합니다!

2024년 1월 27일 (토) 오후 2:44, Chang-Hyeon Hwang (mayaoriginal) <thd...@gmail.com>님이 작성:
--
이 메일은 Google 그룹스 '머신러닝/딥러닝 도서 Q&A' 그룹에 가입한 분들에게 전송되는 메시지입니다.
이 그룹에서 탈퇴하고 더 이상 이메일을 받지 않으려면 ml-dl-book-qn...@googlegroups.com에 이메일을 보내세요.
웹에서 이 토론을 보려면 https://groups.google.com/d/msgid/ml-dl-book-qna/ef516c32-0438-460d-813b-5d3a0d9f0655n%40googlegroups.com을(를) 방문하세요.
Reply all
Reply to author
Forward
0 new messages