파이썬 학습, 평가를 위한 데이터 자동 분할
혼자 공부하려고 정리했어요~ 생성된 데이터를 학습(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) prin..
2022. 11. 22.
