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

구링님의 프로필 이미지
구링

작성한 질문수

현업 실무자에게 배우는 Kaggle 머신러닝 입문 - ML 엔지니어 실무 꿀팁

XGBoost를 이용해서 뇌졸중(Stroke) 발생유무 예측해보기 - Stroke Prediction 데이터셋

categorical object

작성

·

141

0

안녕하세요, 혹시 XGBoost 예제 코드에서
 
일부 범주형 카테고리라고 생각하는게 있는데요
 
work_type
smoking_status 같은 경우는 one-hotencoding 처리를 하는게 아닌가, 여쭤 보고 싶습니다.
 
혹시 원핫 인코딩이 아닌, LabelEncoder 을 하신 이유가 있으실까요?
 

답변 1

0

AISchool님의 프로필 이미지
AISchool
지식공유자

안녕하세요~. 반갑습니다.

말씀해주신대로 범주형값은 work_type, smoking_status 같은 범주형 카테고리는 one-hot encoding 처리를 해주는 것이 좋습니다.

따라서 해당 column들에 대해서는 one-hot encoding을 해주는 것이 더 좋겠네요.

좋은 하루되세요~.

감사합니다.

구링님의 프로필 이미지
구링

작성한 질문수

질문하기