JuJuKwakKwak

Python) 학습과 테스트 나눌 때 본문

Data Science/코드 관련 모음

Python) 학습과 테스트 나눌 때

jujukwakkwak 2022. 8. 5. 08:30 
from sklearn.model_selection import train_test_split

x = liver[liver.columns.difference(['Dataset'])]
y = liver['Dataset']

train_x, test_x, train_y, test_y = train_test_split(x, y, stratify=y, train_size=0.7, random_state=1)

print(train_x.shape, test_x.shape, train_y.shape, test_y.shape)
저작자표시

'Data Science > 코드 관련 모음' 카테고리의 다른 글

python) 시간대별 명사 추출하고 개수 세기  (0) 2022.08.22
python) 특정 유형만 추출하기  (0) 2022.08.22
Python) 빈 데이터프레임에 반복해서 채워넣기  (0) 2022.08.04
jupyter notebook 글씨 예쁘게 바꾸기  (0) 2022.07.29
데이터 전처리  (0) 2022.07.27
'Data Science/코드 관련 모음' Related Articles more