[sklearn] 'stratify' 의 역할(train_test_split) [ic]train_test_split[/ic]에서 [ic]stratify[/ic]가 뭐 하는 녀석인지 헷갈리는가? 당겨보세요이 포스팅은 쿠팡 파트너스 활동의 일환으로, 이에 따른 일정액의 수수료를 제공받습니다. 그렇다면 잘 들어왔다. 이번 포스팅에서는 [ic]stratify[/ic]를 미적용했을 때 어떤 문제가 발생하는지 알아보고, [ic]stratify[/ic]를 통해 문제를 해결해볼 거다. 1) 예제 데이터 준비df_2 = pd.DataFrame({'class_id': ['A', 'A', 'A', 'A', 'A','A' ,'B', 'B', 'B'], 'feature1': [1, 2, 3, 4, 5, 6, 7, 8, 9],}) print(.. 머신러닝,딥러닝 2023. 1. 5. 이전 1 다음