[빅데이터 분석기사] 실기 3회- 2유형 prob

 

대표사진 삭제

 

사진 설명을 입력하세요.

다 잘 해놓고 마지막에서 오류가 났다, 에러를 보니 1차원이 아니란다 그래서 뭔가 싶어서 찾아봄

 

2특정 클래스로 분류될 확률을 구할 경우 (predict_proba)를 사용하는데

y_result_prob = model.predict_proba(x_test) 여기까진 맞음음

 

 

근데 여기서 prob를 좀 더 이해할 필요가 있다

다른 예제를 들면

 

사진 삭제

사진 설명을 입력하세요.

result_prob	=	pd.DataFrame({	
				'result':	y_result,
				'prob_0':	y_result_prob[:,0], 
				'prob_1':	y_result_prob[:,1], 
				'prob_2':	y_result_prob[:,2] 
})

 

y_result_prob에는 분류된 클래스 값 그에 속할 확률값으로 구성된 데이터인거임!!

# Class 0일 확률 : y_result_prob[:,0]

# Class 1일 확률 : y_result_prob[:,1]

# Class 2일 확률 : y_result_prob[:,2]

 

문제에서 travelInsurance가 1일 확률을 구해서 제출하라 아래 식처럼 써야 했던 것이다

y_result_prob	=	model.predict_proba(x_test)
result=pd.DataFrame({'ID':ID, 'proba':y_result_prob[:,1]})
result.to_csv('datafox',index=False)
pd.read_csv("datafox")

 

이러면 오류없이 잘 나온당