RStudio 中绘制 PR 曲线、ROC 曲线和代价曲线

要绘制 PR 曲线、ROC 曲线和代价曲线,你可以使用以下代码在 RStudio 中进行绘制。

# 安装和加载所需的包
install.packages('pROC')
install.packages('PRROC')
install.packages('MLmetrics')
install.packages('ggplot2')
library(pROC)
library(PRROC)
library(MLmetrics)
library(ggplot2)

# 创建一个包含样本序号、真实类别和学习得分的数据框
df <- data.frame(Sample_ID = c(1, 2, 3, 4, 5),
                 True_Class = c(1, 0, 1, 1, 0),
                 Score = c(0.8, 0.6, 0.9, 0.7, 0.4))

# 计算 PR 曲线
pr <- pr.curve(scores.class0 = df$Score, weights.class0 = ifelse(df$True_Class == 0, 1, 0),
               scores.class1 = df$Score, weights.class1 = ifelse(df$True_Class == 1, 1, 0))
pr.plot(pr)

# 计算 ROC 曲线
roc <- roc(df$True_Class, df$Score)
roc.plot(roc)

# 计算代价曲线
costs <- cost_curve(df$True_Class, df$Score)
plot(costs$threshold, costs$cost)

# 添加标签和标题
xlabel <- 'Threshold'
ylabel <- 'Cost'
title <- 'Cost Curve'
xrange <- range(costs$threshold)
yrange <- range(costs$cost)
text(x = xrange[1] + 0.1 * diff(xrange), y = yrange[2] - 0.1 * diff(yrange),
     labels = xlabel, adj = 0)
text(x = xrange[1] + 0.1 * diff(xrange), y = yrange[1] + 0.1 * diff(yrange),
     labels = ylabel, adj = 0)
title(main = title)

上述代码中,首先安装并加载了绘制 PR 曲线、ROC 曲线和代价曲线所需的包。然后,创建了一个包含样本序号、真实类别和学习得分的数据框。接下来,计算并绘制了 PR 曲线、ROC 曲线和代价曲线。最后,添加了标签和标题以增加图表的可读性。

请根据你的实际数据和需求进行适当的修改。

RStudio 中绘制 PR 曲线、ROC 曲线和代价曲线

原文地址: https://www.cveoy.top/t/topic/ZeV 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录