Question

我希望在插入符号库的列车（）之后显示混淆矩阵，但我有些疑惑。＆＃34;火车（）＆＃34;应该在火车上？（我不确定，因为＆＃34;控制＆＃34;参数）。＆＃34;预测（）＆＃34;在测试集上？在整个数据集上进行预测似乎很奇怪......

# df_corpus = Document Term Matrix + 1 column of Cos.code(class which are 203.2.2, 204.3.2 ...)
dataset <- df_corpus
control <- trainControl(method = "repeatedcv", number = 10, repeats = 3)
seed <- 7

metric <- "Accuracy"
preProcess=c("center", "scale")

# Linear Discriminant Analysis
set.seed(seed)
fit.lda <- train(Cos.code~., data=dataset, method="lda", metric=metric,preProc=c("center", "scale"), trControl=control)
ldaClasses <- predict(fit.lda)
cm <- confusionMatrix(data = ldaClasses, dataset$Cos.code)
F1_score(cm$table, "lda")

感谢您的帮助

Answer 1

你可以得到这样的混淆矩阵：

confusionMatrix(predict(fit.lda,dataset$Cos.code),dataset$Cos.code)

您可以以相同的方式为测试集计算混淆矩阵，只需切换数据集即可。

但我相信您的模型应该已包含您想要的信息检查打印这两个对象时给出的信息。

fit.lda

fit.lda$finalModel

R插入符号/混淆矩阵

1 个答案: