인프런 커뮤니티 질문&답변

김준수님의 프로필 이미지
김준수

작성한 질문수

[퇴근후딴짓] 빅데이터 분석기사 실기 (작업형1,2,3)

2유형 X_train, y_train 합칠때

해결된 질문

작성

·

99

0

선생님 강의중 2회 2유형 문제 강의중 X_train 과 y_train 그리고 X_test 까지 총 데이터가 3개가 주어진 문제에서 df = pd.concat([X_train,y_train['Reached.on.Time_Y.N']], axis =1) 로 데이터를 합쳐주셨는데 이러면 검증데이터분리 부분에서 (target = df.pop('Reached.on.Time_Y.N')으로 타겟데이터를 뺐을때)

X_tr, X_val, y_tr, y_val = train_test_split(df, target, test_size = 0.2, random_state = 1) 이렇게 합친 데이터를 사용해서 분리해도 문제가 없을까요?

답변 1

0

퇴근후딴짓님의 프로필 이미지
퇴근후딴짓
지식공유자

X_tr, X_val, y_tr, y_val = train_test_split(df, target, test_size = 0.2, random_state = 1) 이렇게 합친 데이터를 사용해서 분리해도 문제가 없을까요?

 

이 코드는 잘못된 코드입니다.

train이 100개

test가 50개라고 가정한다면

df 는 150개이고, target은 100개 입니다.

수가 맞지 않아 에러가 발생합니다.

김준수님의 프로필 이미지
김준수

작성한 질문수

질문하기