如何从 R 中的 h2o 模型绘制 AUCPR 曲线？答案

【问题标题】：How to plot the AUCPR curve from h2o model in R?如何从 R 中的 h2o 模型绘制 AUCPR 曲线？
【发布时间】：2020-01-13 01:47:35
【问题描述】：

我在 R 中使用 h2o.ai 平台进行了某种数据分析，我想收到模型/预测的 AUCPR 曲线。

我已经尝试使用“PRROC”包，但它发现我的数据集（140 万个实例）无法正常工作或变慢。对于其他可用的软件包，我真的不知道如何从 h2o 模型中提取数据。

pr <- h2o.predict(V_PUF_AGG1_NPI_ALLEX_BINAR.drf.tt.standard, data.test)

给我一个预测矩阵（我可以用它来进一步处理吗？）：

  predict        p1           p2
1       1 0.9999427 5.731940e-05
2       1 0.9999606 3.939748e-05
3       1 0.9999744 2.556443e-05
4       1 0.9999659 3.413081e-05
5       1 0.9999606 3.939748e-05
6       1 0.9999545 4.554749e-05

[987141 rows x 3 columns]

所以我正在寻找绘制 AUCPR 曲线的快速解决方案。

很容易得到ROC曲线，但是没有办法直接从h2o得到AUCPR曲线：

plot(h2o.performance(V_PUF_AGG1_NPI_ALLEX_BINAR.drf.tt.standard, valid=T), type='roc')

谢谢 - 约翰

【问题讨论】：

标签： r plot h2o roc auc

【解决方案1】：

您可以绘制例如的 PRROC测试集的方式如下，其中mod是你的H2O模型对象，test_hf是你的测试集。

library(tidyverse)

perf <- h2o.performance(mod,test_hf)

metrics <- as.data.frame(h2o.metric(perf))

metrics %>%
  ggplot(aes(recall,precision)) + 
  geom_line() +
  theme_minimal()

【讨论】：