r语言 - 误差:响应变量和预测变量必须是相同长度的向量

Question

我遇到以下错误，如何解决？

install.packages("pROC")
library(pROC)
lr_predict <- predict(lg_model,train_data, probability =TRUE)
auc_gbm = roc(test_data$Class,lr.predict,plot = TRUE, col = "blue")

设置级别：控制 = 0，案例 = 1
roc.default(test_data$Class， lr.predict， plot = TRUE， col = "blue"( 中的
    错误：响应和预测变量必须是相同长度的向量。

Answer 1

按照 @Shirin Yavari 的响应，您可以使用这样的代码：

lr_predict <- predict(lg_model, test_data, probability =TRUE)
auc_gbm = roc(test_data$Class, lr.predict, plot = TRUE, col = "blue")