glmnet 범주 별 회귀를 사용하여 0과 1 사이의 확률을 얻는 방법

Question

내 응답 변수는 1에서 7까지 범주 형입니다. glmnet을 사용하여 이해합니다. 유형을 응답으로 설정하고 예측 확률을 얻을 수 있습니다.

prob.vec = predict.cv.glmnet (cvfit,이 newX = x.Test가, S = "lambda.min" 유형 = "응답") 그러나

, 내가 관심 다른 카테고리의 확률도 가지고 있습니다. 이러한 기능이 glmnet에 존재하는지 궁금합니다.

Answer 1

family = 'multinomial'을 cv.glmnet 내에 설정하면 모두 설정해야합니다. 다음은 간단한 예입니다.

library(glmnet) 
y <- iris$Species 
x <- as.matrix(iris[,1:4]) 
m1 <- cv.glmnet(x, y, family = 'multinomial') 

predict(m1, newx = x, s = 'lambda.min', type = 'response')[1:2,,] 
#  setosa versicolor virginica 
#[1,] 0.9992059 0.000794057 2.548623e-20 
#[2,] 0.9961627 0.003837330 1.206001e-18