查看原文
其他

一行代码使用雷达图比较不同模型指标的差异

段湘杰 R语言统计与绘图 2023-10-21

构建多个不同的模型,选择最优的模型,今天来学习一个函数,使用这个函数快速比较不同模型的各项参数,并且一行代码可视化模型指标的差异。

1. 加载R包

加载需要用到的R包。

library(performance)
library(see)
library(survival)
library(tidyverse)

2. 加载数据

使用lung数据集中的数据进行演示,并简单进行数据处理。

dt <- lung %>% 
  na.omit() %>% 
  mutate(
    status = ifelse(status == '1'01),
    sex = factor(sex)
  )
glimpse(dt)

3. 构建模型

m0 <- glm(status ~ age + sex + ph.ecog, data = dt, family = "binomial")
m1 <- glm(status ~ age + pat.karno + ph.ecog, data = dt, family = "binomial")
m2 <- glm(status ~ age + meal.cal + ph.ecog, data = dt, family = "binomial")

4. 比较模型

可以使用compare_performance()函数比较不同的模型。

compare_performance(m0, m1, m2)

如上,输出模型的质量评估指标不同数据。

5. 可视化模型指标差异

可以使用plot()函数可视化模型的不同指标,绘制雷达图。

plot(compare_performance(m0, m1, m2))

参考资料

  1. performance包帮助文件。

《完》


关注下方公众号,分享更多更好玩的R语言知识

觉得有帮助的请点赞、分享、在看走起!

点个在看,SCI马上发表。

您可能也对以下帖子感兴趣

文章有问题?点此查看未经处理的缓存