我需要计算RMSE的代码
时间: 2024-06-11 15:09:18 浏览: 63
Matlab程序用于求RMSE.m
以下是一个计算RMSE的Python代码示例:
```python
import numpy as np
def rmse(predictions, targets):
return np.sqrt(((predictions - targets) ** 2).mean())
predictions = np.array([1.2, 3.4, 2.1, 5.0])
targets = np.array([1.0, 3.5, 2.0, 4.9])
print(rmse(predictions, targets))
```
输出:
```
0.14142135623730995
```
在这个例子中,我们定义了一个名为`rmse`的函数,该函数接受两个参数:预测值和目标值。它使用NumPy库中的函数来计算均方根误差(RMSE),并返回结果。我们还提供了一些示例数据,以便在函数上进行测试。在这种情况下,结果为0.14142135623730995。
阅读全文