我现在有两个向量,请你写一段python代码用来比较他们的RMSE和MAE
时间: 2024-04-08 13:35:19 浏览: 137
当给定两个向量,可以使用以下的Python代码来计算它们的RMSE和MAE:
```python
import numpy as np
def calculate_rmse(vector1, vector2):
squared_error = np.square(np.subtract(vector1, vector2))
mean_squared_error = np.mean(squared_error)
rmse = np.sqrt(mean_squared_error)
return rmse
def calculate_mae(vector1, vector2):
absolute_error = np.abs(np.subtract(vector1, vector2))
mae = np.mean(absolute_error)
return mae
# 示例向量
vector1 = [1, 2, 3, 4, 5]
vector2 = [2, 4, 6, 8, 10]
rmse = calculate_rmse(vector1, vector2)
mae = calculate_mae(vector1, vector2)
print("RMSE:", rmse)
print("MAE:", mae)
```
在上述代码中,我们使用`numpy`库来进行向量计算。`calculate_rmse`函数计算两个向量的均方根误差(RMSE),首先计算两个向量的差值平方,然后求平均值,最后取平方根。`calculate_mae`函数计算两个向量的平均绝对误差(MAE),直接计算两个向量的绝对误差的平均值。最后,我们将RMSE和MAE打印出来。请确保在运行代码之前,将示例向量替换为你要比较的向量。
阅读全文