아래 예측 함수는 -ve 값도 제공하므로 확률이 될 수 없습니다.
param <- list(max.depth = 5, eta = 0.01, objective="binary:logistic",subsample=0.9)
bst <- xgboost(param, data = x_mat, label = y_mat,nround = 3000)
pred_s <- predict(bst, x_mat_s2)
나는 구글 & 시도 pred_s <- predict(bst, x_mat_s2,type="response")
했지만 작동하지 않았다.
질문
대신 확률을 예측하는 방법?
사용한 설정으로 기본적으로 확률이 출력되지 않습니까? 내 말은 : pred_s를 조사한 결과 확률이 아닌 것이 확실합니까?
—
kpb
음수 값이 없습니다. 확률은 0에서 1 사이 여야합니다.
—
GeorgeOfTheRF
명백한 문제가 없습니다. (하지만 파이썬 래퍼에 더 익숙합니다). 당신은 추가 시도
—
역전
outputmargin=F
받는 predict
기능? 가로 outputmargin
설정 T
되면 로지스틱 변환 전에 값을 반환합니다.
Python의 경우 API
—
Anton Tarasenko
predict_proba
에서 구현을 복사 할 수 있습니다 sklearn
. github.com/dmlc/xgboost/blob/master/python-package/xgboost/…