서두르세요! "데이스쿨 40% 할인" 12월 2일까지!
분석시각화 대회 코드 공유 게시물은
내용 확인 후
좋아요(투표) 가능합니다.
삭제된 댓글입니다
[질문] 맨 처음 train과 val 세트를 나눌 때 (아래 코드), 이 때는 Stratified 하게 나눠줄 필요가 없을까요?
train_x, val_x, train_y, val_y = train_test_split(train_x, train_y, test_size=0.2, random_state=42)
Stratified 하게 나누면 더 좋을 것 같습니다.
train_x, val_x, train_y, val_y = train_test_split(train_x, train_y, test_size=0.2, random_state=42, stratify=train_y)
이렇게 하면 될거같아요
답변 감사합니다~
데이콘(주) | 대표 김국진 | 699-81-01021
통신판매업 신고번호: 제 2021-서울영등포-1704호
서울특별시 영등포구 은행로 3 익스콘벤처타워 901호
이메일 dacon@dacon.io | 전화번호: 070-4102-0545
Copyright ⓒ DACON Inc. All rights reserved
많은 도움이 되었습니다. 감사합니다.