用R语言做ROC曲线代码
时间: 2024-03-09 08:40:07 浏览: 98
ROC曲线绘制
以下是在R语言中生成ROC曲线的示例代码:
```
# 读取数据
data <- read.csv("data.csv")
# 将数据集分为训练集和测试集
train_index <- sample(nrow(data), 0.7*nrow(data))
train_data <- data[train_index, ]
test_data <- data[-train_index, ]
# 训练模型
model <- glm(response ~ predictor1 + predictor2, data = train_data, family = binomial)
# 预测测试集
prob <- predict(model, test_data, type = "response")
# 计算ROC曲线
library(pROC)
roc_curve <- roc(test_data$response, prob)
# 绘制ROC曲线
plot(roc_curve, main = "ROC Curve", xlab = "False Positive Rate", ylab = "True Positive Rate")
lines(x = c(0, 1), y = c(0, 1), lty = 2)
```
需要注意的是,上述代码需要将数据集中的响应变量命名为“response”,预测变量命名为“predictor1”和“predictor2”。此外,需要安装pROC包以计算ROC曲线。
阅读全文