当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


R yardstick rsq R 平方


使用相关性计算决定系数。有关 R 平方的传统度量,请参阅 rsq_trad()

用法

rsq(data, ...)

# S3 method for data.frame
rsq(data, truth, estimate, na_rm = TRUE, case_weights = NULL, ...)

rsq_vec(truth, estimate, na_rm = TRUE, case_weights = NULL, ...)

参数

data

data.frame 包含由 truthestimate 参数指定的列。

...

目前未使用。

truth

真实结果的列标识符(即 numeric )。这应该是一个不带引号的列名,尽管此参数是通过表达式传递的并且支持quasiquotation(您可以不带引号的列名)。对于 _vec() 函数,一个 numeric 向量。

estimate

预测结果的列标识符(也是 numeric )。与 truth 一样,可以通过不同的方式指定,但主要方法是使用不带引号的变量名称。对于 _vec() 函数,一个 numeric 向量。

na_rm

logical 值,指示在计算继续之前是否应剥离 NA 值。

case_weights

案例权重的可选列标识符。这应该是一个不带引号的列名称,其计算结果为 data 中的数字列。对于 _vec() 函数,一个数值向量。

tibble 包含列 .metric.estimator.estimate 以及 1 行值。

对于分组 DataFrame ,返回的行数将与组数相同。

对于 rsq_vec() ,单个 numeric 值(或 NA )。

细节

决定系数的两个估计值 rsq()rsq_trad() 的公式不同。前者保证 (0, 1) 上的值,而后者在模型无信息时可能生成不准确的值(请参阅示例)。两者都是一致性/相关性的衡量标准,而不是准确性的衡量标准。

rsq() 只是 truthestimate 之间的平方相关性。

由于rsq() 在内部计算相关性,因此如果truthestimate 为常数,则可能会导致除以零的错误。在这些情况下,会引发警告并返回NA。当模型预测所有样本的单个值时,就会发生这种情况。例如,消除除截距之外的所有预测变量的正则化模型就可以做到这一点。另一个例子是不包含分割的 CART 模型。

参考

克瓦尔塞斯。关于 的注意事项。美国统计学家(1985)卷。 39 (4) 第 279-285 页。

也可以看看

其他数字指标:ccc() , huber_loss_pseudo() , huber_loss() , iic() , mae() , mape() , mase() , mpe() , msd() , poisson_log_loss() , rmse() , rpd() , rpiq() , rsq_trad() , smape()

其他一致性指标:ccc()rpd()rpiq()rsq_trad()

作者

马克斯·库恩

例子

# Supply truth and predictions as bare column names
rsq(solubility_test, solubility, prediction)
#> # A tibble: 1 × 3
#>   .metric .estimator .estimate
#>   <chr>   <chr>          <dbl>
#> 1 rsq     standard       0.879

library(dplyr)

set.seed(1234)
size <- 100
times <- 10

# create 10 resamples
solubility_resampled <- bind_rows(
  replicate(
    n = times,
    expr = sample_n(solubility_test, size, replace = TRUE),
    simplify = FALSE
  ),
  .id = "resample"
)

# Compute the metric by group
metric_results <- solubility_resampled %>%
  group_by(resample) %>%
  rsq(solubility, prediction)

metric_results
#> # A tibble: 10 × 4
#>    resample .metric .estimator .estimate
#>    <chr>    <chr>   <chr>          <dbl>
#>  1 1        rsq     standard       0.874
#>  2 10       rsq     standard       0.879
#>  3 2        rsq     standard       0.891
#>  4 3        rsq     standard       0.916
#>  5 4        rsq     standard       0.892
#>  6 5        rsq     standard       0.858
#>  7 6        rsq     standard       0.873
#>  8 7        rsq     standard       0.852
#>  9 8        rsq     standard       0.915
#> 10 9        rsq     standard       0.884

# Resampled mean estimate
metric_results %>%
  summarise(avg_estimate = mean(.estimate))
#> # A tibble: 1 × 1
#>   avg_estimate
#>          <dbl>
#> 1        0.883
# With uninformitive data, the traditional version of R^2 can return
# negative values.
set.seed(2291)
solubility_test$randomized <- sample(solubility_test$prediction)
rsq(solubility_test, solubility, randomized)
#> # A tibble: 1 × 3
#>   .metric .estimator .estimate
#>   <chr>   <chr>          <dbl>
#> 1 rsq     standard     0.00199
rsq_trad(solubility_test, solubility, randomized)
#> # A tibble: 1 × 3
#>   .metric  .estimator .estimate
#>   <chr>    <chr>          <dbl>
#> 1 rsq_trad standard       -1.01

# A constant `truth` or `estimate` vector results in a warning from
# a divide by zero error in the correlation calculation.
# `NA` will be returned in these cases.
truth <- c(1, 2)
estimate <- c(1, 1)
rsq_vec(truth, estimate)
#> Warning: A correlation computation is required, but `estimate` is constant and has 0 standard deviation, resulting in a divide by 0 error. `NA` will be returned.
#> [1] NA
源代码:R/num-rsq.R

相关用法


注:本文由纯净天空筛选整理自Max Kuhn等大神的英文原创作品 R squared。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。