MSE 均方误差
# MSE计算
def mse(target, predict):
return ((target - predict)**2).mean()
mse_val = mse(np.array(BH_old_data), np.array(BH_new_data))
mse_val_new = mse(np.array(BH_old_data), np.array(new_model_predict))
RMSE 均方根误差
# RMSE计算
def rmse(predictions, targets):
return np.sqrt(((predictions - targets) ** 2).mean())
rmse_val = rmse(np.array(BH_old_data), np.array(BH_new_data))
rmse_val_new = rmse(np.array(BH_old_data), np.array(new_model_predict))
R 相关系数
# R相关系数计算
def R(target, predict):
SSR = sum((predict - (target).mean())**2)
SST = sum((target - (target).mean())**2)
return SSR/SST
r_val = R(np.array(BH_old_data), np.array(BH_new_data))
r_val_new = R(np.array(BH_old_data), np.array(new_model_predict))
MAE 平均绝对误差