R yardstick rmse 均方根误差

计算均方根误差。 rmse() 是与原始数据具有相同单位的度量。

用法

rmse(data, ...)

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

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

参数

data: data.frame 包含由 truth 和 estimate 参数指定的列。
...: 目前未使用。
truth: 真实结果的列标识符(即 numeric )。这应该是一个不带引号的列名，尽管此参数是通过表达式传递的并且支持quasiquotation(您可以不带引号的列名)。对于 _vec() 函数，一个 numeric 向量。
estimate: 预测结果的列标识符(也是 numeric )。与 truth 一样，可以通过不同的方式指定，但主要方法是使用不带引号的变量名称。对于 _vec() 函数，一个 numeric 向量。
na_rm: logical 值，指示在计算继续之前是否应剥离 NA 值。
case_weights: 案例权重的可选列标识符。这应该是一个不带引号的列名称，其计算结果为 data 中的数字列。对于 _vec() 函数，一个数值向量。

值

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

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

对于 rmse_vec() ，单个 numeric 值(或 NA )。

也可以看看

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

其他准确度指标：ccc() , huber_loss_pseudo() , huber_loss() , iic() , mae() , mape() , mase() , mpe() , msd() , poisson_log_loss() , smape()

作者

马克斯·库恩

例子

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

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) %>%
  rmse(solubility, prediction)

metric_results
#> # A tibble: 10 × 4
#>    resample .metric .estimator .estimate
#>    <chr>    <chr>   <chr>          <dbl>
#>  1 1        rmse    standard       0.715
#>  2 10       rmse    standard       0.673
#>  3 2        rmse    standard       0.716
#>  4 3        rmse    standard       0.644
#>  5 4        rmse    standard       0.737
#>  6 5        rmse    standard       0.675
#>  7 6        rmse    standard       0.807
#>  8 7        rmse    standard       0.801
#>  9 8        rmse    standard       0.635
#> 10 9        rmse    standard       0.692

# Resampled mean estimate
metric_results %>%
  summarise(avg_estimate = mean(.estimate))
#> # A tibble: 1 × 1
#>   avg_estimate
#>          <dbl>
#> 1        0.710

源代码：R/num-rmse.R

相关用法

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