데이터 분리할떄

Inflearn Community Q&A

[After-work activities] Big Data Analysis Engineer Practical (Work-type 1,2,3)

Resolved

Written on

293

from sklearn.model_selection import train_test_split

X_tr,X_val,y_tr,y_val = train_test_split(

X_train,

y_train['Reached.on.Time_Y.N'],

test_size=0.2,

random_state=2021

)

이거라면 y_train['타켓값']넣고 통째로 외워도 될까요??

그리고

from sklearn.model_selection import train_test_split

X_tr, X_val, y_tr, y_val = train_test_split(train.drop('TravelInsurance', axis=1),

train['TravelInsurance'],

test_size=0.1,

random_state=1204)

X_tr.shape, X_val.shape, y_tr.shape, y_val.shape

이건 왜 위에와 다르게 train값에 drop값을 넣는건가요?

python머신러닝빅데이터pandas빅데이터분석기사

Answer 1

Instructor

train_test_split 함수에 들어갈 입력 값으로 X(독립변수, 피처), y(종속변수, 타겟)이 있어요
X데이터에 타겟이 있는 경우 제외해야 합니다. 없다면 제외할 필요가 없겠죠
y데이터에는 타겟 컬럼(1개 컬럼)만 필요합니다.

분할 하기 전에 X데이터와 y데이터 샘플을 출력해서 눈으로 확인후 train_test_split 를 실행하는 것을 추천합니다.

무작정 외우는 방식은 시험 1주 전이 아니라면 데이터가 어떤 형식일지 몰라 추천하진 않습니다.

asked