Obtenir les classes prévues de R objet glmnet

Question

Je suis en train de construire simples modèles de régression logistique multi-classe à l'aide glmnet R. Mais lorsque je tente de prédire les données de test et obtenir un tableau de contingence je reçois une erreur. Une séance échantillon est reproduit ci-dessous.

> mat = matrix(1:100,nrow=10)
> test = matrix(1:50,nrow=5)

> classes <- as.factor(11:20)

> model <- glmnet(mat, classes, family="multinomial", alpha=1)
> pred <- predict(model, test)
> table(pred, as.factor(11:15))
  Error in table(pred, as.factor(11:15)) : 
  all arguments must have the same length

Toute aide sera appréciée. R Noob ici.

Merci.

Answer 1

La méthode predict pour un objet glmnet nécessite que vous spécifiez une valeur pour le s argument, ce qui indique que les valeurs du paramètre de régularisation pour lesquels vous souhaitez prévisions.

(glmnet correspond au modèle pour plusieurs valeurs de ce paramètre de régularisation simultanément.)

Donc, si vous ne spécifiez pas de valeur pour s, retourne predict.glmnet prédictions pour tous les valeurs. Si vous voulez juste un ensemble de prévisions, vous devez soit définir une valeur pour s lorsque vous appelez predict, ou vous devez extraire la colonne correspondante après le fait.