在Python中,计算均方误差(Mean Squared Error, MSE)通常使用NumPy库,其基本公式为:
MSE = 1/n * ∑(yi - y_pred)²
其中,`yi` 是真实值,`y_pred` 是预测值,`n` 是数据点的数量。
import numpy as npdef mean_squared_error(y_true, y_pred):return np.mean((y_true - y_pred) 2)示例数据y_true = np.array([1, 2, 3, 4, 5])y_pred = np.array([1.2, 2.5, 3.3, 4.1, 5.8])计算均方误差mse = mean_squared_error(y_true, y_pred)print("Mean Squared Error:", mse)
输出结果为:
Mean Squared Error: 0.
这个函数`mean_squared_error`接受两个参数:`y_true`表示真实值数组,`y_pred`表示预测值数组。计算出的MSE值越小,表示模型的预测越准确。
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 举报,一经查实,本站将立刻删除。
如需转载请保留出处:https://sigusoft.com/bj/68111.html