파이썬 학습, 평가를 위한 데이터 자동 분할

혼자 공부하려고 정리했어요~

 

생성된 데이터를 학습(train), 평가(test)로 자동분할 하는 방법을 정리해봤습니다.

값을 임의로 추출하여 학습할 데이터와 평가할 데이터를 나누어줍니다.

import numpy as np
from sklearn.model_selection import  train_test_split

X = np.arange(20)
y = X **2
print('X:', X, 'y:', y)

X_train, X_test, y_train, y_test = train_test_split(X, y, random_state=0, test_size=0.2)
#random하게 섞지만 0번째 seed로 / 8 train : 2 test로 분할합니다
print("X_train", X_train)
print("y_train", y_train)
print("X_test", X_test)
print("y_test", y_test)