使用 stargazer 从 caret 导出 glm train fit 摘要？

Question

我想使用包 "stargazer" 来导出带插入符号的火车的拟合摘要。

例如

LogReg <- train(
Target ~ .,
data = DecileFramesTrain[[i]], method="glm", family="binomial",
trControl = trainControl(method = "cv", number = 3, verboseIter = TRUE)
)

然后

stargazer(summary(LogReg), title="Coefficients of (binomial) logistic regression fit", align=TRUE, summary=TRUE)

但是，我收到以下回复：

"Error: Unrecognized object type."

如有任何建议，我们将不胜感激！我无法在线找到解决方案。提前致谢，祝您有愉快的一天！

Answer 1

可以获得 caret 生成的最终模型的 stargazer 摘要：

library(caret)
library(mlbench) # for PimaIndiansDiabetes data
library(stargazer)

data(PimaIndiansDiabetes)

LogReg <- train(diabetes ~ ., data=PimaIndiansDiabetes, method="glm",family="binomial", 
                trControl = trainControl(method = "cv", number = 3, verboseIter = TRUE))

glm1 <- glm(diabetes ~ ., data=PimaIndiansDiabetes, family="binomial")
LogReg$finalModel$call <- glm1$call
stargazer(LogReg$finalModel)

stargazer 函数使用 call 参数来检查模型类型。最终插入符号模型中缺少它。 glm call 参数属于调用 class。有关详细信息，请参阅 R 会话中的 ?call。上面代码中的倒数第二行将 call 参数添加到最终插入符模型。

注意 1：严格来说，由于插入符号交叉验证，添加到最终插入符号模型的 call 参数不正确。

注2：我没有查看stargazer制作的LaTeX代码。

使用 stargazer 从 caret 导出 glm train fit 摘要？

Export glm train fit summary from caret with stargazer?

model

r

summary

stargazer

r-caret