2017-09-12 2 views
0

私の応答変数は1から7までのカテゴリです。glmnetの使用を理解しますタイプを応答に設定して予測の確率を得ることができます。glmnetカテゴリカル回帰を使用して0と1の確率を得る方法

prob.vec = predict.cv.glmnet(cvfit、newx = X.test、S = "lambda.min"、 タイプ= "応答")

しかし、私は興味があります他のカテゴリーの確率も有する。そのような機能がglmnetに存在するのだろうかと思います。

答えて

0

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 
関連する問題