인프런 영문 브랜드 로고
인프런 영문 브랜드 로고

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

hello4298님의 프로필 이미지
hello4298

작성한 질문수

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

6회 기출유형 (작업형2)

xgboost 사용 시 오류

해결된 질문

작성

·

247

0

  • xgboost로 학습시킬때 오류가 나오던데요.

  • xgboost는 y값이 숫자형으로 되어있어야 학습이 가능한건지 궁금합니다.

해당 문제에는 target 값이 low/high/... 와 같이 str 형식이여서 xgboost가 사용이 불가한건지해서요 :)

답변 1

0

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

네 맞습니다. xgboost는 타겟이 숫자로 되어 있어야 하며

숫자도 1,2,3,4는 안됩니다.

0, 1, 2, 3 형태로 0부터 시작해야 합니다.

따라서 부스팅 계열의 모델을 사용하고 싶다면 덜 까다로운 lightGBM을 추천합니다. (마지막에 뽀너스 영상 추가 해뒀어요)

hello4298님의 프로필 이미지
hello4298
질문자

감사합니다. 보통 분류에는 xgboost가 결과값이 좋아서 항상 같이 체크했었는데 lightGBM을 활용하겠습니다 ㅎㅎ!

hello4298님의 프로필 이미지
hello4298

작성한 질문수

질문하기