Python中的mean_squared_error()函数用于计算预测误差
发布时间:2024-01-19 16:41:08
在Python中,mean_squared_error()函数是scikit-learn库中的一个函数,用于计算预测误差。
该函数的用法如下:
from sklearn.metrics import mean_squared_error mean_squared_error(y_true, y_pred, squared=True)
其中,y_true是真实的目标值,y_pred是预测的目标值,squared是一个布尔值,表示是否返回平方的误差,默认为True。
下面是一个例子,展示如何使用mean_squared_error()函数来计算预测误差:
from sklearn.metrics import mean_squared_error
import numpy as np
# 真实的目标值
y_true = np.array([2, 4, 6, 8])
# 预测的目标值
y_pred = np.array([1, 3, 5, 7])
# 计算预测误差
mse = mean_squared_error(y_true, y_pred)
# 打印结果
print("Mean Squared Error:", mse)
输出结果为:
Mean Squared Error: 2.0
上述例子中,我们定义了真实的目标值y_true为[2, 4, 6, 8],预测的目标值y_pred为[1, 3, 5, 7],然后使用mean_squared_error()函数计算出了预测误差mse。根据计算结果,预测误差为2.0。
需要注意的是,mean_squared_error()函数计算的是预测误差的平方,如果希望得到未平方的预测误差,可以将squared参数设置为False。例如:
mse = mean_squared_error(y_true, y_pred, squared=False)
希望上述解答能对你有所帮助!
