기억 저장소

클라우드 기반 인공지능 개발과 DevOps 실무

인공지능/머신러닝

머신 러닝 : Training / Test set , 트레이닝 테스트 용 나누기 , 데이터 나누기

하늘.98 2021. 11. 24. 17:35

 

 

 

transform 한  X , y 를 가져온다. 

from sklearn.model_selection import train_test_split

import를 불러온 뒤 

X_train, X_test , y_train , y_test = train_test_split(X,y,test_size=0.2,random_state=3)

아래의 식 처럼 입해준다.

test_size =       => X_test, y_test 를 얼마나 저장할지 단위를 나타내주며

random_state=              => random 무작위로 뽑아오는 숫자를 뜻한다

                                         같은 식으로 같은 숫자를 대입하면 값은 값이 나올 수 있다.