RMSE和MAE计算的python代码
时间: 2024-03-02 11:23:58 浏览: 63
python之MSE、MAE、RMSE的使用
5星 · 资源好评率100%
以下是计算 RMSE 和 MAE 的 Python 代码:
```python
import numpy as np
# 预测值和真实值
y_pred = np.array([1.2, 2.1, 3.3, 4.0, 5.1])
y_true = np.array([1.0, 2.0, 3.5, 4.2, 5.0])
# 计算 MSE 和 MAE
mse = np.mean((y_pred - y_true) ** 2)
mae = np.mean(np.abs(y_pred - y_true))
# 计算 RMSE
rmse = np.sqrt(mse)
print("RMSE: {:.3f}".format(rmse))
print("MAE: {:.3f}".format(mae))
```
当然,这只是一个示例代码,实际应用中需要根据具体情况调整代码。
阅读全文