-
Data를 train 과 test set으로 원하는 비율만큼 분리하는법data-science-summary/summary 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
반응형'data-science-summary > summary' 카테고리의 다른 글
선형회귀(LinearRegression) in python (0) 2020.09.17 상관분석 (Correlation) in python (0) 2020.09.16 PCA in python (0) 2020.09.16 Robust scaling 하는법 in python (0) 2020.09.16 z-score scaling 하는법 in python (0) 2020.09.14