data-science-summary/summary
Data를 train 과 test set으로 원하는 비율만큼 분리하는법
고수트
2020. 9. 16. 21:58
반응형
<엔지니어는 구현을 못하면 모르는것이다>
Data를 train 과 test set으로 원하는 비율만큼 분리하려면 sklearn 의 train_test_split 모듈을 이용한다.
# train/test set 분리
import numpy as np
from sklearn.model_selection import train_test_split
X, y = np.arange(10).reshape((5, 2)), range(5)
# X, y 값 다있을때
# 7:3 으로 분기
train_x, test_x, train_y, test_y = train_test_split(X, y, test_size=0.3, random_state=123)
# X 값만 있을때
train, test = train_test_split(X, test_size=0.3, random_state=123)
train
반응형