Programming/python

python train test set split 분리

Jink_23 2023. 10. 20. 15:16
728x90
반응형

 

 

scikit-learn 미설치 상태면 설치가 필요합니다

 

 

train_test_split 사용 시

 - X, y 인자는 필수

 - test_size : test data size 지정. (Default : 0.25)

 - random_state : 지정 시 동일 random 주소에서 시작 → 다른 환경해서 해도 동일한 값 추출

 - stratify : y class가 label 일 때, 사용. y 선택 시,  train 과 test Data의 y 값 비를 동일하게 가져감

 

 

 

 

 

 

 

 

728x90
반응형