• 카테고리

    질문 & 답변
  • 세부 분야

    자격증 (데이터 사이언스)

  • 해결 여부

    해결됨

lightgbm 결과(y, pred)

23.12.02 07:27 작성 조회수 208

0

안녕하세요 선생님!

lightgbm cliassifier를 사용할때

범주형 변수를 타입만 category로 바꾸면 사용이 가능하다고 하셨는데요

학습할 결과도 범주형 그대로 학습시키면

출력되는값도 그대로 범주형으로 출력되나요?

 

예를들어 예측해야할 값이 A,B,C,D,E라면 그대로 넣으면

결과 = 모델.predict(test데이터)

 

했을때 결과를 출력하면 A,B,C,D,E로 나오는지 궁금합니다.

 

 

답변 1

답변을 작성해보세요.

0

네 타겟은 카테고리 자료형이 아니더라도

범주형(문자)은 범주형(문자)으로 결과가 나옵니다 :)