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

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

hjwhjw00님의 프로필 이미지
hjwhjw00

작성한 질문수

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

pred_proba 질문

해결된 질문

작성

·

219

0

데이터를 예측할 때 predict가 아니라 predict.proba를 사용하였을 때,

데이터프레임을 제작할 때 pred_proba[:,1]로 작성하는 것을 외웠지만 [:,1]을 작성하는 이유를 모르겠어서 질문 드립니다!

답변 2

1

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

pred_proba[행,열]입니다.

행의 모든 데이터 중에서 0은 첫번째 컬럼, 1은 두번째 컬럼이에요

0과 1이 있다면

첫번째 컬럼은 0 확률값이고, 두번째 컬럼은 1확률값입니다.

1확률값을 뽑기 위함입니다.

1

타겟이 두번째 열에 해당하는 데이터여서 그 데이터만 추출하려고

hjwhjw00님의 프로필 이미지
hjwhjw00

작성한 질문수

질문하기