从 R 包标准修改 `pr_curve` 和 `auc_curve` 的图形大小

Question

我正在尝试使用库 "yardstick" 生成 ROC 曲线和精确回忆曲线。但是，我找不到修改图形形状的方法。这是一个玩具示例。

## Precision-recall curve
data.frame(true = as.factor(rep(c(0,1), 10)), 
           pred = runif(20)) %>% 
  pr_curve(truth = true, pred) %>% 
  autoplot()

## ROC curve
data.frame(true = as.factor(rep(c(0,1), 10)), 
           pred = runif(20)) %>% 
  roc_curve(truth = true, pred) %>% 
  autoplot()

当您运行代码时，生成的图形如下所示；

上图（ROC曲线）是正方形，下图（precision-recall曲线）是矩形。

我试过

更改 pdf 函数中的 width 和 height 选项
更改 ggplot2 支持的不同选项（例如 plot.margin 使用 theme）

但没找到让两个图形形状相同的好方法。

如何统一它们的形状（或形式）？

任何意见将不胜感激。

Answer 1

来自 ggplot2 的

coord_fixed() 就可以了。请注意，如果您希望绘图区域为正方形，还需要调整 xlim 和 ylim。

pr_curve(tmp1, truth = true, pred) %>% 
    autoplot() +
    coord_fixed(xlim = 0:1, ylim = 0:1)

从 R 包标准修改 `pr_curve` 和 `auc_curve` 的图形大小

Modify figure sizes of `pr_curve` and `auc_curve` from R package yardstick

r

ggplot2

yardstick