[머신러닝] train_test_split (데이터 나누기)

 

 

 

머신러닝 공부를 하면서 scikit-learn 라이브러리를 사용하게 되었다.

오늘은 그 안에서도 데이터를 분리해주는 모듈인 train_test_split에 대해서 알아보고자 한다.

 

 

import

from sklearn.model_selection import train_test_split

 

option

x_train, x_test, y_train, y_test = train_test_split(x_data, y_data)

위와 같이 작성하면 랜덤으로 train data와 test data를 분리해준다.

 

train_test_split(x,y, test_size = 0.3, train_size = 0.7)

위와 같이 데이터의 비율을 조절할 수도 있다. (test 30% / train 70%)

 

train_test_split(x,y, shuffle=False)

shuffle = False이면 랜덤 없이 순차적으로 데이터를 분리한다.