欢迎访问宙启技术站
智能推送

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)

希望上述解答能对你有所帮助!