Python中通过mean_squared_error()函数计算预测模型的均方误差
发布时间:2024-01-19 16:39:43
在Python中,可以使用sklearn.metrics模块中的mean_squared_error函数来计算预测模型的均方误差。该函数的使用方法如下:
from sklearn.metrics import mean_squared_error
# 创建实际值和预测值的数组
actual_values = [2, 4, 6, 8, 10]
predicted_values = [1.5, 3.5, 5.5, 7.5, 9.5]
# 使用mean_squared_error函数计算均方误差
mse = mean_squared_error(actual_values, predicted_values)
# 打印均方误差
print("均方误差为:", mse)
上述代码中,我们首先导入了mean_squared_error函数,并创建了实际值和预测值的数组。然后,使用mean_squared_error函数来计算均方误差。最后,使用print函数打印出计算得到的均方误差。
在本例中,实际值为[2, 4, 6, 8, 10],预测值为[1.5, 3.5, 5.5, 7.5, 9.5],我们可以通过计算均方误差来评估模型的准确性。在这个例子中,均方误差为0.25,预测模型的准确性相对较高。
均方误差是评估预测与实际观测值之间的误差的常用指标。它计算预测值与实际值之间的平方差的平均值。均方误差越小,预测模型的准确性越高。
除了计算均方误差,sklearn.metrics模块还提供了许多其他常用的评估函数,例如均值绝对误差(mean_absolute_error)和决定系数(r2_score)。可以根据具体的需求选择适合的评估函数。
