计算R方指标和均方根误差(RMSE)的自定义函数(R语言)
R方指标(R-squared)和均方根误差(Root Mean Square Error,RMSE)是评估回归模型性能常用的指标。本文将介绍如何使用R语言编写自定义函数来计算这两个指标。
首先,我们将创建一个名为"calculate_r_squared_rmse"的函数,该函数接受两个参数:观测值(observed)和预测值(predicted)。以下是完整的函数代码:
calculate_r_squared_rmse <- function(observed, predicted) {
# 计算平均观测值
mean_observed <- mean(observed)
# 计算总平方和(Total Sum of Squares,TSS)
tss <- sum((observed - mean_observed)^2)
# 计算残差平方和(Residual Sum of Squares,RSS)
rss <- sum((observed - predicted)^2)
# 计算R方指标
r_squared <- 1 - rss / tss
# 计算均方根误差(RM