[python] predict vs predict_proba의 차이

• predict() 는 범주의 예측을 반환 
• predict_proba()는 확률을 반환

1) predict() 

: predict의 경우, 각각 모델의 최종 예측값을 출력 

• 회귀모델 : 특정 값 출력 ( 0.56 , 0.65 .. ) 
• 분류모델 : 가장 확률이 높은 클래스 출력 (0 ,1 ,2 .. )  

2) predict_proba() 

: 분류모델이 각 클래스에 대한 확률을 출력 

• 회귀에서는 사용할 수 없다 

model = GradientBoostingClassifier(random_state = 0 ,
                                   learning_rate = 0.01,
                                   n_estimators = 450)
model.fit(x_train , y_train)
#predict
y_pred = model.predict(x_test)
#pred_proba
y_pred_proba = model.predict_proba(x_test)

 

실행결과  

y_pred

#실행결과
array([0, 1, 0, ..., 0, 0, 0])

y_pred_proba

#실행결과
array([[0.95087811, 0.04912189],
       [0.22272886, 0.77727114],
       [0.90919185, 0.09080815],
       ...,
       [0.80303484, 0.19696516],
       [0.96121441, 0.03878559],
       [0.9648995 , 0.0351005 ]])