python实现两个矩阵数组的mse计算

在Python中，你可以使用NumPy库来计算两个矩阵数组之间的均方误差(Mean Squared Error，MSE)。以下是基本步骤：

import numpy as np

def mse(matrix1, matrix2):
    # 确保矩阵都是numpy数组
    matrix1 = np.array(matrix1)
    matrix2 = np.array(matrix2)

    # 计算每个元素的差值平方
    diff_squares = (matrix1 - matrix2) ** 2

    # 求和并除以总元素数（这里假设矩阵是对角线以上的元素有效）
    n = matrix1.size
    sum_of_diff_squares = np.sum(diff_squares)

    # 返回均方误差
    return sum_of_diff_squares / n

# 示例矩阵
matrix_a = [[1, 2], [3, 4]]
matrix_b = [[1.1, 2.05], [3.01, 4.09]]

mse_value = mse(matrix_a, matrix_b)
print("Mean Squared Error:", mse_value)