본문 바로가기
파이썬. 머신러닝/sklearn

train_test_split에서 stratify 역할

by 한국수달보호협회장 2024. 9. 21.

 

 

stratifyt - train 샘플을 추출할때 비율을 맞게 뽑아주는 역할을 한다

 

 

https://yeko90.tistory.com/entry/what-is-stratify-in-traintestsplit

 

[sklearn] 'stratify' 의 역할(train_test_split)

[ic]train_test_split[/ic]에서 [ic]stratify[/ic]가 뭐 하는 녀석인지 헷갈리는가? 그렇다면 잘 들어왔다. 이번 포스팅에서는 [ic]stratify[/ic]를 미적용했을 때 어떤 문제가 발생하는지 알아보고, [ic]stratify[/ic]

yeko90.tistory.com