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

파란비님의 프로필 이미지
파란비

작성한 질문수

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

lightgbm 결과(y, pred)

해결된 질문

작성

·

278

0

안녕하세요 선생님!

lightgbm cliassifier를 사용할때

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

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

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

 

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

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

 

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

 

 

답변 1

0

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

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

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

파란비님의 프로필 이미지
파란비

작성한 질문수

질문하기