y를 pop

Question

from sklearn.model_selection import train_test_split X_tr, X_val, y_tr, y_val = train_test_split(train.drop('output', axis=1), train['output'], test_size=0.15, random_state=2022) 이런식으로 안 하고 전처리 과정에서 y = train.pop으로 해도 되나요?

퇴근후딴짓 · Answer

네!! 그렇게 하는 것이 훨씬 깔깜할 것 같아요 전처리 시작하면 y = train.pop로 먼저 진행하시죠!

인프런 AI 인턴 · Answer

안녕하세요, 인프런 AI 인턴입니다. 김서현님. 질문 주신 내용은 데이터 프레임에서 타겟 변수를 분리하는 방식과 관련이 있습니다. 전처리 과정에서 y = train.pop('output') 을 사용해도 괜찮습니다. 이는 pop 함수를 통해 &lsquo;output&rsquo; 컬럼을 train 데이터 프레임에서 제거하고, 그 컬럼을 y 변수에 할당하는 방식입니다. train.pop('output') 사용의 이점 : pop 은 특정 열을 데이터 프레임에서 분리하면서 바로 다른 변수에 담기 때문에, 전처리 단계에서 자주 사용됩니다. 이후 모델 학습 단계에서 train 에는 독립 변수만 남게 되어, model.fit(X_train, y_train) 을 수행할 준비가 됩니다. train_test_split 사용 : 반대로, train_test_split 을 이용하면 데이터를 분리하면서 동시에 학습(validation) 세트와 테스트(test) 세트를 생성할 수 있습니다. 이 방식은 특정 데이터셋을 분리하여 검증 세트를 형성할 때 유리합니다. 다음 관련된 질문 링크를 참고하시면 더 명확한 이해에 도움이 될 수 있을 것입니다: 데이터 전처리에서 y 값을 pop 함수를 사용해 분리하는 이유 저는 질문자님의 이해를 돕고자 지식공유자님의 기존 답변을 참고하여 유사한 답변 링크를 추천해드리고 있습니다. 현재 베타 기능으로 답변이 만족스럽지 않을 수 있는 점 양해 부탁드립니다. 🙏 추가적으로 궁금한 점이 있으시면, 이어서 질문해 주세요. 곧 지식공유자께서 답변해 주실 것입니다.