강의

멘토링

로드맵

Inflearn brand logo image

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

김라의님의 프로필 이미지
김라의

작성한 질문수

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

[뽀너스] lightGBM 잘 활용하기🔥

검증데이터 분할

작성

·

44

0

검증데이터 분할을 할 때 왜 gender를 drop하나요?

from sklearn.model_selection import train_test_split

X_train,X_val,y_train,y_val =train_test_split(train.drop('gender',axis=1),train['gender'],test_size=0.2,random_state=0)

print(X_train.shape,X_val.shape,y_train.shape,y_val.shape)

답변 2

0

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

타겟을 분리하려고 해요!

  1. train.drop('gender',axis=1),train['gender'] 방식을 사용하거나

  2. 미리 pop()을 사용해서 gender를 target으로 옮겨놓고 target = train.pop('gender')


    train, target으로 작성할 수 있어요

 

그럼 lightgbm이어도

target=train.pop("gender)를 활용해서 데이터 분할 해도 되는건가요?

X_train, X_val, y_train, y_val = train_test_split(train. target, test_size=0.2, random_state=0)

0

안녕하세요, 인프런 AI 인턴입니다. 김라의님께서 문의하신 검증 데이터 분할에서 왜 ‘gender’ 변수를 드롭하는지에 대해 설명해드리겠습니다.

검증 데이터 분리의 이유:

  • 독립변수와 종속변수 분리: 머신러닝 모델을 학습할 때, 데이터는 일반적으로 독립변수(피처)와 종속변수(타겟)로 나뉩니다. 타겟 변수는 예측하고자 하는 값인데, 이 값은 학습할 데이터로부터 독립적으로 관리되어야 합니다.
  • ‘gender’는 예측하려는 타겟 변수로 간주되며, 모델이 입력 데이터에서 다른 피처를 기반으로 ‘gender’를 예측하도록 해야 합니다.

코드 설명:

  • train.drop('gender', axis=1): 이 부분은 ‘gender’ 열(타겟)을 제거하고 나머지 피처들을 입력 데이터 세트로 사용하기 위한 것입니다.
  • train['gender']: 이는 ‘gender’ 열(타겟 변수)을 y 데이터로 지정합니다.
  • train_test_split 함수는 데이터를 훈련 세트와 검증 세트로 분할합니다. 이 과정에서, 독립변수와 종속변수를 명확히 분리하는 것이 중요합니다.

더 많은 맥락을 이해하고 싶으시다면 아래의 유사한 질문과 그에 대한 답변을 참조하세요:
- 데이터 분리할때: 이 질문에서는 비슷하게 타겟 변수와 피처를 분리하는 과정에 대한 설명이 있습니다.

저는 질문자님의 이해를 돕고자 지식공유자님의 기존 답변을 참고하여 유사한 답변 링크를 추천해드리고 있습니다. 현재 베타 기능으로 답변이 만족스럽지 않을 수 있는 점 양해 부탁드립니다. 🙏 추가적으로 궁금한 점이 있으시면, 이어서 질문해 주세요. 곧 지식공유자께서 답변해 주실 것입니다.

김라의님의 프로필 이미지
김라의

작성한 질문수

질문하기