작성
·
44
답변 2
0
타겟을 분리하려고 해요!
train.drop('gender',axis=1),train['gender'] 방식을 사용하거나
미리 pop()을 사용해서 gender를 target으로 옮겨놓고 target = train.pop('gender')
train, target으로 작성할 수 있어요
0
안녕하세요, 인프런 AI 인턴입니다. 김라의님께서 문의하신 검증 데이터 분할에서 왜 ‘gender’ 변수를 드롭하는지에 대해 설명해드리겠습니다.
train.drop('gender', axis=1)
: 이 부분은 ‘gender’ 열(타겟)을 제거하고 나머지 피처들을 입력 데이터 세트로 사용하기 위한 것입니다.train['gender']
: 이는 ‘gender’ 열(타겟 변수)을 y 데이터로 지정합니다.train_test_split
함수는 데이터를 훈련 세트와 검증 세트로 분할합니다. 이 과정에서, 독립변수와 종속변수를 명확히 분리하는 것이 중요합니다.더 많은 맥락을 이해하고 싶으시다면 아래의 유사한 질문과 그에 대한 답변을 참조하세요:
- 데이터 분리할때: 이 질문에서는 비슷하게 타겟 변수와 피처를 분리하는 과정에 대한 설명이 있습니다.
저는 질문자님의 이해를 돕고자 지식공유자님의 기존 답변을 참고하여 유사한 답변 링크를 추천해드리고 있습니다. 현재 베타 기능으로 답변이 만족스럽지 않을 수 있는 점 양해 부탁드립니다. 🙏 추가적으로 궁금한 점이 있으시면, 이어서 질문해 주세요. 곧 지식공유자께서 답변해 주실 것입니다.
그럼 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)