Невозможно указать тип = «ответ» в функции предсказания Каре

Question

Я пытаюсь построить кривую ROC для моей модели регрессии glmnet. Для того, чтобы сделать это, я пытаюсь предсказать, используя тип = «ответ» в предсказать функцию:

pred_glmnet_s10_2class <- predict(model_train_glmnet_s10_2class, 
            newdata=testing_s10_2class, 
            s = "model_train_glmnet_s10_2class$finalModel$lambdaOpt", 
            type="response") 

и я получаю следующее сообщение об ошибке:

Error in predict.train(model_train_glmnet_s10_2class, newdata = testing_s10_2class, : type must be either "raw" or "prob"

Мои предсказания и класс метки являются двоичными 0 и 1 и были учтены. Любая помощь действительно ценится. Кроме того, любые идеи о том, как построить AUC (Area Under ROC curve) и количество функций? Благодаря!

Answer 1

Предполагая, что model_train_glmnet_s10_2class был сгенерирован train (с указанием кода будет полезно) ...

Использование predict(model_train_glmnet_s10_2class) использует predict.train и использует оптимальные значения лямбда определяются train автоматически. Если вам нужны вероятности, просто используйте type = "prob".

Ваш синтаксис согласуется с predict.glmnet, а не predict.train.

Как сказано в документации, это действительно плохая идея использовать model_train_glmnet_s10_2class$finalModel непосредственно делать предсказания