- 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 ]])
'파이썬(Python)' 카테고리의 다른 글
[python] matplotlib 한글 자음 모음 분리 현상 (0) | 2024.01.09 |
---|---|
[웹크롤링] window.scrollTo(0, document.body.scrollHeight) (0) | 2024.01.07 |
[python] - 일시정지(sleep) 함수 사용 (1) | 2023.11.25 |
[python] 최댓값과 최솟값의 초깃값 설정 (0) | 2023.11.14 |
[Python] BeautifulSoup으로 HTML 파싱 (0) | 2023.11.07 |