[英]How to plot a ROC curve for a SVM model in R
我已经使用以下代码训练和测试了一个模型
library(e1071)
library(readxl)
library(caret)
class1.svm.model <- svm(Class ~ ., data = class1.trainset,cost=20, cross=10,type="C-classification",kernel="radial",na.action=na.omit)
class1.svm.pred <- predict(class1.svm.model, class1.testset)
finalmatrix<-data.matrix(class1.svm.pred, rownames.force = F)
test<-table(pred = class1.svm.pred, true = class1.testset[,c(15768)])
confusionMatrix(test)
但无法为模型绘制 ROC 曲线。 请帮助我使用正确的语法绘制 ROC 曲线以查看我的测试数据的性能。
用类绘制 SVM 的 ROC 曲线:
roc_svm_test <- roc(response = class1.trainset$Class, predictor =as.numeric(class1.svm.pred))
plot(roc_svm_test, add = TRUE,col = "red", print.auc=TRUE, print.auc.x = 0.5, print.auc.y = 0.3)
legend(0.3, 0.2, legend = c("test-svm"), lty = c(1), col = c("blue"))
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.